RegNetY-6.4GF_dds_8gpu.yaml

MODEL:
  TYPE: regnet
  NUM_CLASSES: 1000
REGNET:
  SE_ON: True
  DEPTH: 25
  W0: 112
  WA: 33.22
  WM: 2.27
  GROUP_W: 72
OPTIM:
  LR_POLICY: cos
  BASE_LR: 0.4
  MAX_EPOCH: 100
  MOMENTUM: 0.9
  WEIGHT_DECAY: 5e-5
  WARMUP_EPOCHS: 5
TRAIN:
  DATASET: imagenet
  IM_SIZE: 224
  BATCH_SIZE: 512
TEST:
  DATASET: imagenet
  IM_SIZE: 256
  BATCH_SIZE: 400
NUM_GPUS: 8
OUT_DIR: .