11 Commits

Author	SHA1	Message	Date
anxiangsir	6a8e3bc917	fixed torch1.9	2021-07-23 15:27:19 +08:00
anxiang	a7605c7e03	added docs	2021-07-14 23:36:14 +08:00
AnXiang	f3c2fa5302	fixed warmup	2021-07-14 12:01:21 +08:00
AnXiang	95b10ed2ca	updated configs	2021-07-07 18:35:13 +08:00
AnXiang	f7007b6729	Resnet2060! ... updated gradient checkpoint for training resnet2060	2021-06-23 13:34:48 +08:00
shiquanyu	33ca5ebc6b	fixed error while change embedding size	2021-05-31 09:50:06 +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	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	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