mirror of https://github.com/JDAI-CV/fast-reid.git
28 lines
433 B
YAML
28 lines
433 B
YAML
_BASE_: "Base-Strongerbaseline.yml"
|
|
|
|
MODEL:
|
|
META_ARCHITECTURE: 'MGN'
|
|
|
|
FREEZE_LAYERS: ["backbone", "b1", "b2", "b3",]
|
|
|
|
BACKBONE:
|
|
WITH_NL: False
|
|
|
|
HEADS:
|
|
NAME: "ReductionHead"
|
|
POOL_LAYER: "maxpool"
|
|
REDUCTION_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
|
|
|