fast-reid/configs/Base-MGN.yml

26 lines
368 B
YAML
Raw Normal View History

_BASE_: "Base-SBS.yml"
2020-07-06 16:48:14 +08:00
MODEL:
META_ARCHITECTURE: 'MGN'
2020-07-06 16:48:14 +08:00
FREEZE_LAYERS: ["backbone", "b1", "b2", "b3",]
BACKBONE:
WITH_NL: False
2020-09-02 18:01:07 +08:00
HEADS:
2020-09-10 10:58:56 +08:00
EMBEDDING_DIM: 256
LOSSES:
NAME: ("CrossEntropyLoss", "TripletLoss",)
CE:
EPSILON: 0.1
2020-09-01 16:16:40 +08:00
SCALE: 1.0
TRI:
MARGIN: 0.0
HARD_MINING: True
NORM_FEAT: False
2020-09-01 16:16:40 +08:00
SCALE: 1.0