vitb14_cxr.yaml

# this corresponds to the CXR config
train:
  batch_size_per_gpu: 40  # For nodes with v100s (32 GB), use 20.
  saveckp_every_n_epoch: 25
student:
  arch: vit_base
  block_chunks: 4
  patch_size: 14
  drop_block_rate: 0.00
  drop_path_rate: 0.30
teacher:
  warmup_teacher_temp_epochs: 50
optim:
  epochs: 100
  warmup_epochs: 5
  base_lr: 0.001
evaluation:
  eval_period_iterations: 300
  tasks:  # from the metadata.csv file of the CANDID processed dataset
  - pneumothorax
crops:
    global_crops_size: 518
    local_crops_size: 196
    global_crops_scale:
    - 0.50
    - 1.00
    local_crops_number: 8
    local_crops_scale:
    - 0.20
    - 0.50
pretrained: true