mirror of
https://github.com/JDAI-CV/fast-reid.git
synced 2025-06-03 14:50:47 +08:00
c3ac4f504c
AMP in partial-fc needs to be done only on backbone; In order to impl `resume training`, need to save & load different part of classifier weight in each GPU.
15 lines
198 B
YAML
15 lines
198 B
YAML
_BASE_: face_base.yml
|
|
|
|
MODEL:
|
|
|
|
BACKBONE:
|
|
DEPTH: 50x
|
|
FEAT_DIM: 25088 # 512x7x7
|
|
DROPOUT: 0.
|
|
|
|
HEADS:
|
|
PFC:
|
|
ENABLED: True
|
|
|
|
OUTPUT_DIR: projects/FastFace/logs/pfc0.1_insightface
|