fast-reid/configs/MSMT17/sbs_R50-ibn.yml

27 lines
441 B
YAML

_BASE_: "../Base-Strongerbaseline.yml"
MODEL:
BACKBONE:
WITH_IBN: True
PRETRAIN_PATH: "/export/home/lxy/.cache/torch/checkpoints/resnet50_ibn_a.pth.tar"
HEADS:
NUM_CLASSES: 1041
DATASETS:
NAMES: ("MSMT17",)
TESTS: ("MSMT17",)
SOLVER:
MAX_ITER: 28000
DELAY_ITERS: 14000
WARMUP_ITERS: 4700
FREEZE_ITERS: 4700
TEST:
EVAL_PERIOD: 5000
PRECISE_BN:
DATASET: 'MSMT17'
OUTPUT_DIR: "logs/msmt17/sbs_R50-ibn"