fast-reid/configs/DukeMTMC/mgn_R50-ibn.yml
liaoxingyu 18a33f7962 feat: add MGN model
support MGN architecture and training config
2020-05-15 11:39:54 +08:00

15 lines
266 B
YAML

_BASE_: "../Base-MGN.yml"
MODEL:
BACKBONE:
PRETRAIN_PATH: "/home/liaoxingyu2/lxy/.cache/torch/checkpoints/resnet50_ibn_a.pth.tar"
HEADS:
NUM_CLASSES: 702
DATASETS:
NAMES: ("DukeMTMC",)
TESTS: ("DukeMTMC",)
OUTPUT_DIR: "logs/dukemtmc/mgn_R50-ibn"