| PRINT_FREQ: 100 | |
| SAVE_FREQ: 1 | |
| MODEL_EMA: True | |
| DATA: | |
| IMG_SIZE: 224 | |
| NUM_WORKERS: 6 | |
| MODEL: | |
| TYPE: revcol_small | |
| NAME: revcol_small | |
| LABEL_SMOOTHING: 0.1 | |
| REVCOL: | |
| INTER_SUPV: True | |
| SAVEMM: True | |
| FCOE: 3.0 | |
| CCOE: 0.7 | |
| DROP_PATH: 0.3 | |
| TRAIN: | |
| EPOCHS: 300 | |
| BASE_LR: 1e-3 | |
| WARMUP_EPOCHS: 20 | |
| WEIGHT_DECAY: 0.05 | |
| WARMUP_LR: 1e-5 | |
| MIN_LR: 1e-6 | |
| OPTIMIZER: | |
| NAME: 'adamw' | |
| CLIP_GRAD: 0.0 | |
| AUG: | |
| COLOR_JITTER: 0.4 | |
| # Use AutoAugment policy. "v0" or "original" | |
| AUTO_AUGMENT: 'rand-m9-mstd0.5-inc1' | |
| # Random erase prob | |
| REPROB: 0.25 | |
| # Random erase mode | |
| REMODE: 'pixel' | |
| # Random erase count | |
| RECOUNT: 1 | |
| # Mixup alpha, mixup enabled if > 0 | |
| MIXUP: 0.8 | |
| # Cutmix alpha, cutmix enabled if > 0 | |
| CUTMIX: 1.0 | |
| # Cutmix min/max ratio, overrides alpha and enables cutmix if set | |
| CUTMIX_MINMAX: None | |
| # Probability of performing mixup or cutmix when either/both is enabled | |
| MIXUP_PROB: 1.0 | |
| # Probability of switching to cutmix when both mixup and cutmix enabled | |
| MIXUP_SWITCH_PROB: 0.5 | |
| # How to apply mixup/cutmix params. Per "batch", "pair", or "elem" | |
| MIXUP_MODE: 'batch' | |