fast-reid/configs/softmax_triplet.yml

43 lines
602 B
YAML
Raw Normal View History

2019-01-10 18:39:31 +08:00
MODEL:
2019-08-13 13:52:25 +08:00
BACKBONE: "resnet50"
2019-01-10 18:39:31 +08:00
INPUT:
2019-04-21 13:38:55 +08:00
SIZE_TRAIN: [256, 128]
SIZE_TEST: [256, 128]
2019-08-07 16:54:50 +08:00
FLIP_PROB: 0.5 # random horizontal flip
2019-01-10 18:39:31 +08:00
PADDING: 10
DATASETS:
NAMES: ("market1501",)
2019-08-13 13:52:25 +08:00
TEST_NAMES: "market1501"
2019-01-10 18:39:31 +08:00
DATALOADER:
SAMPLER: 'triplet'
2019-01-10 18:39:31 +08:00
NUM_INSTANCE: 4
SOLVER:
2019-08-20 09:36:47 +08:00
OPT: 'adam'
LOSSTYPE: ('softmax', 'triplet')
2019-04-23 12:56:38 +08:00
MAX_EPOCHS: 150
2019-01-10 18:39:31 +08:00
BASE_LR: 0.00035
WEIGHT_DECAY: 0.0005
WEIGHT_DECAY_BIAS: 0.0005
IMS_PER_BATCH: 64
2019-01-10 18:39:31 +08:00
2019-04-23 12:56:38 +08:00
STEPS: [40, 90]
2019-01-10 18:39:31 +08:00
GAMMA: 0.1
WARMUP_FACTOR: 0.01
WARMUP_ITERS: 10
2019-07-29 20:57:19 +08:00
EVAL_PERIOD: 30
2019-01-10 18:39:31 +08:00
TEST:
2019-04-21 13:38:55 +08:00
IMS_PER_BATCH: 512
2019-01-10 18:39:31 +08:00
WEIGHT: "path"
OUTPUT_DIR: "logs/market/batch256/"
2019-01-10 18:39:31 +08:00