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.
11 lines
198 B
Python
11 lines
198 B
Python
# encoding: utf-8
|
|
"""
|
|
@author: xingyu liao
|
|
@contact: sherlockliao01@gmail.com
|
|
"""
|
|
|
|
from .modeling import *
|
|
from .config import add_face_cfg
|
|
from .trainer import FaceTrainer
|
|
from .datasets import *
|