Commit Graph

10 Commits

Author SHA1 Message Date
anxiang
0cc88e2a24 Simplify running commands(single node and multi nodes)
1. Update training logs(glint360k)
2. Update install docs
3. Fix distributed training
2021-03-21 15:07:33 +08:00
AnXiang
873e94c50f Update README.md 2021-03-19 12:36:15 +08:00
AnXiang
cc4d662302 FIX BUG
1. model parallel x_grad rescale
2021-03-15 18:55:11 +08:00
anxiang
89a9c5df4c Update README.md 2021-03-12 22:50:27 +08:00
Jia Guo
2ae7221fc4 Update README.md 2021-03-12 21:32:48 +08:00
AnXiang
88b0739fc2 Update Speed Benchmark. 2021-03-12 21:29:18 +08:00
AnXiang
f89ecaaa54 Update pretrain model.
1. Add training log and pretrain
2021-03-12 16:22:15 +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
dc98ba3536 Update experiments. 2021-03-09 17:24:29 +08:00
AnXiang
ede87b4a49 Add arcface_torch. 2021-03-08 21:00:07 +08:00