Commit Graph

5 Commits

Author SHA1 Message Date
AnXiang
66739ca6b8 updated partial fc doc string. 2021-07-13 16:13:27 +08:00
anxiang
c41b8a79a3 Support Export Onnx
1. Update default num_workers
2. Fix resume partial fc
2021-05-09 13:28:42 +08:00
AnXiang
cc4d662302 FIX BUG
1. model parallel x_grad rescale
2021-03-15 18:55:11 +08:00
AnXiang
d8738240ca Add FP16 Support.
Merge PartialFC pytorch into arcface_torch.
Pytorch1.6+ is all you need
2021-03-11 13:13:01 +08:00
AnXiang
ede87b4a49 Add arcface_torch. 2021-03-08 21:00:07 +08:00