2019-01-10 18:39:31 +08:00
|
|
|
MODEL:
|
2019-07-29 20:57:19 +08:00
|
|
|
PRETRAIN_PATH: '/export/home/lxy/.cache/torch/checkpoints/resnet50-19c8e357.pth'
|
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-01-10 18:39:31 +08:00
|
|
|
PROB: 0.5 # random horizontal flip
|
|
|
|
|
PADDING: 10
|
|
|
|
|
|
|
|
|
|
DATASETS:
|
2019-07-31 21:11:54 +08:00
|
|
|
NAMES: ("market1501",)
|
2019-01-10 18:39:31 +08:00
|
|
|
|
|
|
|
|
DATALOADER:
|
|
|
|
|
SAMPLER: 'softmax_triplet'
|
|
|
|
|
NUM_INSTANCE: 4
|
|
|
|
|
|
|
|
|
|
SOLVER:
|
|
|
|
|
OPTIMIZER_NAME: 'Adam'
|
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
|
2019-07-31 21:11:54 +08:00
|
|
|
IMS_PER_BATCH: 256
|
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"
|
|
|
|
|
|
2019-07-31 21:11:54 +08:00
|
|
|
OUTPUT_DIR: "logs/market/batch256/"
|
2019-01-10 18:39:31 +08:00
|
|
|
|
|
|
|
|
|
