fast-reid/configs/Base-bagtricks.yml

77 lines
1.1 KiB
YAML
Raw Normal View History

MODEL:
2021-01-18 11:36:38 +08:00
META_ARCHITECTURE: Baseline
BACKBONE:
2021-01-18 11:36:38 +08:00
NAME: build_resnet_backbone
NORM: BN
DEPTH: 50x
LAST_STRIDE: 1
2020-09-10 10:58:56 +08:00
FEAT_DIM: 2048
WITH_IBN: False
PRETRAIN: True
HEADS:
2021-01-18 11:36:38 +08:00
NAME: EmbeddingHead
NORM: BN
2020-09-10 10:58:56 +08:00
WITH_BNNECK: True
POOL_LAYER: GlobalAvgPool
2021-01-18 11:36:38 +08:00
NECK_FEAT: before
CLS_LAYER: Linear
LOSSES:
2020-07-06 16:48:14 +08:00
NAME: ("CrossEntropyLoss", "TripletLoss",)
CE:
EPSILON: 0.1
SCALE: 1.
TRI:
MARGIN: 0.3
HARD_MINING: True
2020-07-06 16:48:14 +08:00
NORM_FEAT: False
SCALE: 1.
INPUT:
2021-01-18 11:36:38 +08:00
SIZE_TRAIN: [ 256, 128 ]
SIZE_TEST: [ 256, 128 ]
REA:
ENABLED: True
PROB: 0.5
FLIP:
ENABLED: True
PADDING:
ENABLED: True
DATALOADER:
SAMPLER_TRAIN: NaiveIdentitySampler
NUM_INSTANCE: 4
2020-07-06 16:48:14 +08:00
NUM_WORKERS: 8
SOLVER:
AMP:
ENABLED: True
2021-01-18 11:36:38 +08:00
OPT: Adam
MAX_EPOCH: 120
BASE_LR: 0.00035
WEIGHT_DECAY: 0.0005
2021-06-01 11:51:22 +08:00
WEIGHT_DECAY_NORM: 0.0005
IMS_PER_BATCH: 64
2021-01-18 11:36:38 +08:00
SCHED: MultiStepLR
STEPS: [ 40, 90 ]
GAMMA: 0.1
WARMUP_FACTOR: 0.1
WARMUP_ITERS: 2000
CHECKPOINT_PERIOD: 30
TEST:
2020-07-06 16:48:14 +08:00
EVAL_PERIOD: 30
IMS_PER_BATCH: 128
CUDNN_BENCHMARK: True