Commit Graph

9 Commits

Author SHA1 Message Date
Mohammad Mahdi Samei
2728b1f68d Update WandB Imports 2023-01-01 19:58:29 +03:30
Mohammad Mahdi Samei
2f64881ad1 Add WandB to arcface_torch.
WandB will log train and validation results. Default configs are in recognition/arcface_torch/configs/base.py
2023-01-01 17:34:26 +03:30
rhonlai
2ee0f53bdf add resume,support adamw 2022-04-30 18:55:20 +08:00
AnXiang
bb221e6e6d updated for WebFace42M
updated readability of the code
2022-01-14 17:48:51 +08:00
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