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
234 B
Python
11 lines
234 B
Python
# encoding: utf-8
|
|
"""
|
|
@author: xingyu liao
|
|
@contact: sherlockliao01@gmail.com
|
|
"""
|
|
|
|
from .partial_fc import PartialFC
|
|
from .face_baseline import FaceBaseline
|
|
from .face_head import FaceHead
|
|
from .iresnet import build_iresnet_backbone
|