fast-reid/fastreid/modeling/backbones/regnet/regnetx/RegNetX-32GF_dds_8gpu.yaml

27 lines
362 B
YAML
Raw Normal View History

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
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: .