RegNetX-12GF_dds_8gpu.yaml

MODEL:
  TYPE: regnet
  NUM_CLASSES: 1000
REGNET:
  DEPTH: 19
  W0: 168
  WA: 73.36
  WM: 2.37
  GROUP_W: 112
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: .