mirror of
https://github.com/JDAI-CV/fast-reid.git
synced 2025-06-03 14:50:47 +08:00
883fd4aede
Summary: Add configurable decorator which can call `Baseline` with `Baseline(cfg)` or `Baseline(cfg, heads=heads, ...)` Decouple linear and loss computation for partial-fc support. Reviewed By: l1aoxingyu
20 lines
285 B
YAML
20 lines
285 B
YAML
_BASE_: Base-bagtricks.yml
|
|
|
|
MODEL:
|
|
BACKBONE:
|
|
WITH_NL: True
|
|
|
|
HEADS:
|
|
POOL_LAYER: GeneralizedMeanPooling
|
|
|
|
LOSSES:
|
|
NAME: ("CrossEntropyLoss", "TripletLoss")
|
|
CE:
|
|
EPSILON: 0.1
|
|
SCALE: 1.0
|
|
|
|
TRI:
|
|
MARGIN: 0.0
|
|
HARD_MINING: False
|
|
SCALE: 1.0
|