fast-reid/configs/Base-MGN.yml
2020-09-10 10:58:56 +08:00

26 lines
381 B
YAML

_BASE_: "Base-Strongerbaseline.yml"
MODEL:
META_ARCHITECTURE: 'MGN'
FREEZE_LAYERS: ["backbone", "b1", "b2", "b3",]
BACKBONE:
WITH_NL: False
HEADS:
EMBEDDING_DIM: 256
LOSSES:
NAME: ("CrossEntropyLoss", "TripletLoss",)
CE:
EPSILON: 0.1
SCALE: 1.0
TRI:
MARGIN: 0.0
HARD_MINING: True
NORM_FEAT: False
SCALE: 1.0