Commit Graph

116 Commits

Author SHA1 Message Date
Xiang An
fa2316589a Update README.md 2024-06-20 23:14:15 +08:00
Samidhya Sarker
ccfc89c77a Update prepare_custom_dataset.md (#2389) 2023-08-02 10:25:12 +08:00
Parsa
9097cd7ba3 Update losses.py
Incorrect `scale` parameter name for ArcFace class.
2023-06-10 18:36:19 +02:00
An Xiang
ab27f4d60f Fixed torch1.12 api changes
remove old train.py and partial_fc.py
2023-04-08 13:12:59 +08:00
XiangAn
5eba1df909 Update documentation for single GPU training
The documentation for training with a single GPU has been updated to reflect the latest best practices and recommendations. This update will help ensure that users have the information they need to effectively train their models using a single GPU. The updated documentation provides guidance on the limitations of using a single GPU, as well as tips for optimizing performance.
2023-02-10 14:36:47 +08:00
XiangAn
ee91084705 Updated docs Fixed issue #2176 2023-02-08 23:45:53 +08:00
XiangAn
e0fdff81ff Resolved torchrun Bug: Fixed issue #2163
Updated torch.distributed.launch to torchrun.
2023-02-08 23:33:07 +08:00
XiangAn
bf32ec2d3f Resolved torchrun Bug: Fixed issue #2163
Updated torch.distributed.launch to torchrun.
2023-02-08 23:15:46 +08:00
XiangAn
b974acc104 Resolved DALI Bug: Fixed issue #2235
DALI's inability to read InsightFace style rec by implementing the script 'scripts/shuffle_rec.py' to generate shuffled recs.
2023-02-08 21:39:03 +08:00
Xiang An
c7bf2048e8 Update README.md 2023-02-04 19:57:44 +08:00
Xiang An
0dee3bdac5 Update prepare_webface42m.md 2023-02-04 19:41:28 +08:00
Xiang An
54f1a95155 Update prepare_webface42m.md 2023-02-04 19:40:24 +08:00
Xiang An
9c9cabf63a Update README.md 2023-02-04 19:37:04 +08:00
Jia Guo
f091989568 fix typo of wandb 2023-01-24 22:33:39 +08:00
Jia Guo
cc64315087 fix using_wandb 2023-01-05 18:31:49 +08:00
Mohammad Mahdi Samei
53b4b2f2cd Update Main Installation Doc 2023-01-03 11:42:10 +03:30
Mohammad Mahdi Samei
a97b8da8e8 Update Dali Doc. 2023-01-03 11:33:24 +03:30
Mohammad Mahdi Samei
9ddacc1e7c Dali installation added and Doc Updated. 2023-01-03 11:21:42 +03:30
Mohammad Mahdi Samei
2398746bcf WebFace Doc Updated 2023-01-03 11:20:41 +03:30
Mohammad Mahdi Samei
90cac16d1a Update WandB Configs, Add Resume Option. 2023-01-03 11:01:58 +03:30
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
Jia Guo
04da49fbbd Update README.md 2022-12-20 20:54:55 +08:00
XiangAn
657ae30e41 fixed arcface impl 2022-12-15 22:44:20 +08:00
XiangAn
5c9f7c47c8 fixed killing 2022-12-15 22:43:24 +08:00
Jia Guo
eaeab8947c Update README.md 2022-11-24 00:10:34 +08:00
Xiang An
babb9a58bb Update README.md 2022-07-13 17:17:42 +08:00
Xiang An
6b349d83b6 Update README.md 2022-07-13 17:15:26 +08:00
Xiang An
f1d524f3cb Update README.md 2022-07-13 17:13:00 +08:00
XiangAn
4f227ba5b4 updated gradient accumulation 2022-07-13 15:52:22 +08:00
Xiang An
e78eee59ca Update README.md 2022-06-10 09:31:19 +08:00
XiangAn
f2fe557b34 fixed 2022-05-31 20:02:15 +08:00
Jack Sigmoid
81dde23084 added random seed control code (#2008)
* added random seed control code

Co-authored-by: Jack_Chen <jiaozhu@orbbec.com>
Co-authored-by: XiangAn <anxiangsir@outlook.com>
2022-05-22 19:29:15 +08:00
Xiang An
29c011eabc Update README.md 2022-05-16 15:59:42 +08:00
Xiang An
83b4782ec3 Update README.md 2022-05-16 15:56:11 +08:00
Xiang An
6d46c84872 Update README.md 2022-05-16 15:55:34 +08:00
Xiang An
0ab4cb790a Update README.md 2022-05-16 15:55:17 +08:00
Xiang An
92a0bb6b0f updated resume 2022-05-14 13:39:56 +08:00
Xiang An
88828a8356 Merge pull request #1990 from rong2010/master
add resume
2022-05-14 13:31:38 +08:00
Xiang An
b5958bc1d8 Update torch2onnx.py
For ViT
2022-05-03 14:10:09 +08:00
rhonlai
2f96de1d38 add resume,change callback log 2022-05-02 16:54:00 +08:00
Xiang An
92c6bd9a4e updated prepare_webface42m.md 2022-05-02 15:51:02 +08:00
Xiang
4df192b12b updated 2022-05-01 16:32:10 +08:00
rhonlai
2ee0f53bdf add resume,support adamw 2022-04-30 18:55:20 +08:00
rhonlai
dfa209dee2 add resume 2022-04-28 11:28:50 +08:00
dream
21bc9e9444 Update prepare_webface42m.md 2022-04-23 10:46:32 +08:00
dream
83300599d9 Update prepare_webface42m.md 2022-04-23 10:46:03 +08:00
dream
d82ceaaedd Update prepare_webface42m.md 2022-04-23 10:45:00 +08:00
Xiang
d9cb10e6bd updated configs
fixed gradient checkpoint
added TODO
updated README.md
2022-04-22 18:24:14 +08:00
JiankangDeng
eca1d9a6cd Update README.md 2022-04-01 11:09:34 +01:00