scripts/image_classification/config/swin_base_patch4_window7_224.yaml [3:64]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
DDP_CONFIG:
  WORLD_SIZE: 1
  WORLD_RANK: 0
  GPU_WORLD_SIZE: 8
  GPU_WORLD_RANK: 0
  DIST_URL: 'tcp://172.31.74.47:23456'
  WOLRD_URLS: ['172.31.74.47']
  AUTO_RANK_MATCH: True
  DIST_BACKEND: 'nccl'
  GPU: 0
  DISTRIBUTED: True

CONFIG:
  TRAIN:
    EPOCH_NUM: 300
    BATCH_SIZE: 128
    LR: 5e-4
    MIN_LR: 5e-6
    WEIGHT_DECAY: 0.05
    USE_WARMUP: True
    WARMUP_EPOCHS: 20
    WARMUP_START_LR: 5e-7
    LR_SCHEDULER:
      NAME: 'cosine'
      DECAY_EPOCHS: 30
      DECAY_RATE: 0.1
    OPTIMIZER:
      NAME: adamw
      EPS: 1e-8
      BETAS: (0.9, 0.999)
      MOMENTUM: 0.9

  AUG:
    COLOR_JITTER: 0.4
    AUTO_AUGMENT: 'rand-m9-mstd0.5-inc1'
    LABEL_SMOOTHING: 0.1
    REPROB: 0.25
    REMODE: 'pixel'
    RECOUNT: 1
    MIXUP: 0.8
    CUTMIX: 1.0
    CUTMIX_MINMAX: None
    MIXUP_PROB: 1.0
    MIXUP_SWITCH_PROB: 0.5
    MIXUP_MODE: 'batch'
    REPEATED_AUG: True

  VAL:
    BATCH_SIZE: 128
    FREQ: 5

  DATA:
    TRAIN_DATA_PATH: '/home/ubuntu/data/imagenet/train/'
    VAL_DATA_PATH:  '/home/ubuntu/data/imagenet/val/'
    NUM_CLASSES: 1000
    IMG_SIZE: 224
    CROP_PCT: 0.9
    TEST_NUM_CROP: 1
    IMAGENET_DEFAULT_NORMALIZE: True
    INTERPOLATION: 'bicubic'

  MODEL:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



scripts/image_classification/config/swin_large_patch4_window7_224_22kto1k.yaml [3:64]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
DDP_CONFIG:
  WORLD_SIZE: 1
  WORLD_RANK: 0
  GPU_WORLD_SIZE: 8
  GPU_WORLD_RANK: 0
  DIST_URL: 'tcp://172.31.74.47:23456'
  WOLRD_URLS: ['172.31.74.47']
  AUTO_RANK_MATCH: True
  DIST_BACKEND: 'nccl'
  GPU: 0
  DISTRIBUTED: True

CONFIG:
  TRAIN:
    EPOCH_NUM: 300
    BATCH_SIZE: 128
    LR: 5e-4
    MIN_LR: 5e-6
    WEIGHT_DECAY: 0.05
    USE_WARMUP: True
    WARMUP_EPOCHS: 20
    WARMUP_START_LR: 5e-7
    LR_SCHEDULER:
      NAME: 'cosine'
      DECAY_EPOCHS: 30
      DECAY_RATE: 0.1
    OPTIMIZER:
      NAME: adamw
      EPS: 1e-8
      BETAS: (0.9, 0.999)
      MOMENTUM: 0.9

  AUG:
    COLOR_JITTER: 0.4
    AUTO_AUGMENT: 'rand-m9-mstd0.5-inc1'
    LABEL_SMOOTHING: 0.1
    REPROB: 0.25
    REMODE: 'pixel'
    RECOUNT: 1
    MIXUP: 0.8
    CUTMIX: 1.0
    CUTMIX_MINMAX: None
    MIXUP_PROB: 1.0
    MIXUP_SWITCH_PROB: 0.5
    MIXUP_MODE: 'batch'
    REPEATED_AUG: True

  VAL:
    BATCH_SIZE: 128
    FREQ: 5

  DATA:
    TRAIN_DATA_PATH: '/home/ubuntu/data/imagenet/train/'
    VAL_DATA_PATH:  '/home/ubuntu/data/imagenet/val/'
    NUM_CLASSES: 1000
    IMG_SIZE: 224
    CROP_PCT: 0.9
    TEST_NUM_CROP: 1
    IMAGENET_DEFAULT_NORMALIZE: True
    INTERPOLATION: 'bicubic'

  MODEL:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



