18 Commits

Author	SHA1	Message	Date
007gzs	f4f91a0dd8	remove fp16 param	2021-05-17 13:49:54 +08:00
007gzs	ddf4f885a7	add main to convert pytorch model to onnx	2021-05-17 10:05:07 +08:00
AnXiang	9b986852c2	added docs for eval	2021-05-14 19:41:08 +08:00
AnXiang	1b857b4400	added eval ijbc for onnx model	2021-05-14 19:33:17 +08:00
AnXiang	127869cca6	Update tutorial for pytorch.	2021-05-12 20:47:57 +08:00
AnXiang	ce6c303a11	Update tutorial for pytorch.	2021-05-12 15:02:40 +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	d4f217b007	Set ms1m-retinaface-t1 default	2021-05-08 17:39:17 +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	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