| train_file: "SIIM-CLS/siim-acr-pneumothorax/stage_1_train_images.csv" |
| valid_file: "SIIM-CLS/siim-acr-pneumothorax/stage_1_test_images.csv" |
| test_file: "SIIM-CLS/siim-acr-pneumothorax/stage_1_test_images.csv" |
|
|
| image_res: 224 |
| batch_size: 64 |
| test_batch_size: 64 |
| num_classes: 1 |
| temp: 0.07 |
| mlm_probability: 0.15 |
| queue_size: 8192 |
| momentum: 0.995 |
| alpha: 0.4 |
| percentage: 0.01 |
|
|
| optimizer: {opt: adamW, lr: 1e-5, weight_decay: 0.02} |
| schedular: {sched: cosine, lr: 1e-5, epochs: 1000, min_lr: 1e-5, decay_rate: 1, warmup_lr: 1e-5, warmup_epochs: 50, cooldown_epochs: 0} |
|
|
