2020-07-17 19:13:45 +08:00
|
|
|
MODEL:
|
|
|
|
TYPE: regnet
|
|
|
|
NUM_CLASSES: 1000
|
|
|
|
REGNET:
|
2020-07-17 19:32:36 +08:00
|
|
|
DEPTH: 23
|
|
|
|
W0: 320
|
|
|
|
WA: 69.86
|
|
|
|
WM: 2.0
|
|
|
|
GROUP_W: 168
|
2020-07-17 19:13:45 +08:00
|
|
|
OPTIM:
|
|
|
|
LR_POLICY: cos
|
2020-07-17 19:32:36 +08:00
|
|
|
BASE_LR: 0.2
|
2020-07-17 19:13:45 +08:00
|
|
|
MAX_EPOCH: 100
|
|
|
|
MOMENTUM: 0.9
|
|
|
|
WEIGHT_DECAY: 5e-5
|
2021-01-22 11:17:21 +08:00
|
|
|
WARMUP_ITERS: 5
|
2020-07-17 19:13:45 +08:00
|
|
|
TRAIN:
|
|
|
|
DATASET: imagenet
|
|
|
|
IM_SIZE: 224
|
2020-07-17 19:32:36 +08:00
|
|
|
BATCH_SIZE: 256
|
2020-07-17 19:13:45 +08:00
|
|
|
TEST:
|
|
|
|
DATASET: imagenet
|
|
|
|
IM_SIZE: 256
|
2020-07-17 19:32:36 +08:00
|
|
|
BATCH_SIZE: 200
|
2020-07-17 19:13:45 +08:00
|
|
|
NUM_GPUS: 8
|
|
|
|
OUT_DIR: .
|