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.
19 lines
321 B
Python
19 lines
321 B
Python
# encoding: utf-8
|
|
"""
|
|
@author: xingyu liao
|
|
@contact: sherlockliao01@gmail.com
|
|
"""
|
|
|
|
from fastreid.config import CfgNode as CN
|
|
|
|
|
|
def add_face_cfg(cfg):
|
|
_C = cfg
|
|
|
|
_C.DATASETS.REC_PATH = ""
|
|
|
|
_C.MODEL.BACKBONE.DROPOUT = 0.
|
|
|
|
_C.MODEL.HEADS.PFC = CN({"ENABLED": False})
|
|
_C.MODEL.HEADS.PFC.SAMPLE_RATE = 0.1
|