fast-reid/projects/FastFace/fastface/modeling
liaoxingyu c3ac4f504c Support amp and resume training in fastface
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.
2021-05-31 17:30:43 +08:00
..
__init__.py Support amp and resume training in fastface 2021-05-31 17:30:43 +08:00
face_baseline.py Support amp and resume training in fastface 2021-05-31 17:30:43 +08:00
face_head.py Support amp and resume training in fastface 2021-05-31 17:30:43 +08:00
iresnet.py Support amp and resume training in fastface 2021-05-31 17:30:43 +08:00
partial_fc.py Support amp and resume training in fastface 2021-05-31 17:30:43 +08:00