2020-04-29 16:18:54 +08:00
|
|
|
MODEL:
|
2021-01-18 11:36:38 +08:00
|
|
|
META_ARCHITECTURE: Baseline
|
2020-04-29 16:18:54 +08:00
|
|
|
|
|
|
|
BACKBONE:
|
2021-01-18 11:36:38 +08:00
|
|
|
NAME: build_resnet_backbone
|
|
|
|
NORM: BN
|
|
|
|
DEPTH: 50x
|
2020-04-29 16:18:54 +08:00
|
|
|
LAST_STRIDE: 1
|
2020-09-10 10:58:56 +08:00
|
|
|
FEAT_DIM: 2048
|
2020-04-29 16:18:54 +08:00
|
|
|
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
|
2021-03-30 15:47:14 +08:00
|
|
|
POOL_LAYER: GlobalAvgPool
|
2021-01-18 11:36:38 +08:00
|
|
|
NECK_FEAT: before
|
2021-03-23 12:10:06 +08:00
|
|
|
CLS_LAYER: Linear
|
2020-04-29 16:18:54 +08:00
|
|
|
|
|
|
|
LOSSES:
|
2020-07-06 16:48:14 +08:00
|
|
|
NAME: ("CrossEntropyLoss", "TripletLoss",)
|
2020-04-29 16:18:54 +08:00
|
|
|
|
|
|
|
CE:
|
|
|
|
EPSILON: 0.1
|
2020-05-01 09:04:51 +08:00
|
|
|
SCALE: 1.
|
2020-04-29 16:18:54 +08:00
|
|
|
|
|
|
|
TRI:
|
2020-04-29 21:06:57 +08:00
|
|
|
MARGIN: 0.3
|
|
|
|
HARD_MINING: True
|
2020-07-06 16:48:14 +08:00
|
|
|
NORM_FEAT: False
|
2020-05-01 09:04:51 +08:00
|
|
|
SCALE: 1.
|
2020-04-29 16:18:54 +08:00
|
|
|
|
|
|
|
INPUT:
|
2021-01-18 11:36:38 +08:00
|
|
|
SIZE_TRAIN: [ 256, 128 ]
|
|
|
|
SIZE_TEST: [ 256, 128 ]
|
2021-03-31 17:07:19 +08:00
|
|
|
|
2020-04-29 16:18:54 +08:00
|
|
|
REA:
|
|
|
|
ENABLED: True
|
|
|
|
PROB: 0.5
|
2021-03-31 17:07:19 +08:00
|
|
|
|
|
|
|
FLIP:
|
|
|
|
ENABLED: True
|
|
|
|
|
2021-03-26 20:17:39 +08:00
|
|
|
PADDING:
|
|
|
|
ENABLED: True
|
2020-04-29 16:18:54 +08:00
|
|
|
|
|
|
|
DATALOADER:
|
2021-03-26 20:17:39 +08:00
|
|
|
SAMPLER_TRAIN: NaiveIdentitySampler
|
2020-04-29 16:18:54 +08:00
|
|
|
NUM_INSTANCE: 4
|
2020-07-06 16:48:14 +08:00
|
|
|
NUM_WORKERS: 8
|
2020-04-29 16:18:54 +08:00
|
|
|
|
|
|
|
SOLVER:
|
2021-03-26 20:17:39 +08:00
|
|
|
AMP:
|
|
|
|
ENABLED: True
|
2021-01-18 11:36:38 +08:00
|
|
|
OPT: Adam
|
2020-12-07 14:19:20 +08:00
|
|
|
MAX_EPOCH: 120
|
2020-04-29 16:18:54 +08:00
|
|
|
BASE_LR: 0.00035
|
|
|
|
WEIGHT_DECAY: 0.0005
|
2021-06-01 11:51:22 +08:00
|
|
|
WEIGHT_DECAY_NORM: 0.0005
|
2020-04-29 21:06:57 +08:00
|
|
|
IMS_PER_BATCH: 64
|
2020-04-29 16:18:54 +08:00
|
|
|
|
2021-01-18 11:36:38 +08:00
|
|
|
SCHED: MultiStepLR
|
|
|
|
STEPS: [ 40, 90 ]
|
2020-04-29 16:18:54 +08:00
|
|
|
GAMMA: 0.1
|
|
|
|
|
2020-12-07 14:19:20 +08:00
|
|
|
WARMUP_FACTOR: 0.1
|
2021-01-22 11:17:21 +08:00
|
|
|
WARMUP_ITERS: 2000
|
2020-04-29 16:18:54 +08:00
|
|
|
|
2020-12-07 14:19:20 +08:00
|
|
|
CHECKPOINT_PERIOD: 30
|
2020-04-29 16:18:54 +08:00
|
|
|
|
|
|
|
TEST:
|
2020-07-06 16:48:14 +08:00
|
|
|
EVAL_PERIOD: 30
|
|
|
|
IMS_PER_BATCH: 128
|
2020-04-29 16:18:54 +08:00
|
|
|
|
|
|
|
CUDNN_BENCHMARK: True
|