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}