mirror of https://github.com/JDAI-CV/fast-reid.git
41 lines
615 B
YAML
41 lines
615 B
YAML
MODEL:
|
|
PRETRAIN_PATH: '/export/home/lxy/.cache/torch/checkpoints/resnet50-19c8e357.pth'
|
|
|
|
|
|
INPUT:
|
|
SIZE_TRAIN: [256, 128]
|
|
SIZE_TEST: [256, 128]
|
|
PROB: 0.5 # random horizontal flip
|
|
PADDING: 10
|
|
|
|
DATASETS:
|
|
NAMES: ("market1501",)
|
|
|
|
DATALOADER:
|
|
SAMPLER: 'softmax_triplet'
|
|
NUM_INSTANCE: 4
|
|
|
|
SOLVER:
|
|
OPTIMIZER_NAME: 'Adam'
|
|
MAX_EPOCHS: 150
|
|
BASE_LR: 0.00035
|
|
WEIGHT_DECAY: 0.0005
|
|
WEIGHT_DECAY_BIAS: 0.0005
|
|
IMS_PER_BATCH: 256
|
|
|
|
STEPS: [40, 90]
|
|
GAMMA: 0.1
|
|
|
|
WARMUP_FACTOR: 0.01
|
|
WARMUP_ITERS: 10
|
|
|
|
EVAL_PERIOD: 30
|
|
|
|
TEST:
|
|
IMS_PER_BATCH: 512
|
|
WEIGHT: "path"
|
|
|
|
OUTPUT_DIR: "logs/market/batch256/"
|
|
|
|
|