Commit Graph

5 Commits

Author SHA1 Message Date
AnXiang
3d42efa397 updated code format 2021-07-07 18:35:27 +08:00
AnXiang
f7007b6729 Resnet2060!
updated gradient checkpoint for training resnet2060
2021-06-23 13:34:48 +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
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