52 Commits

Author	SHA1	Message	Date
itsnine	be357873d3	Fixed CPU inference for onnx arcface	2021-09-15 18:34:37 +05:00
Vinh Quang Tran	aee09e3ccf	Remove unused code	2021-08-18 14:31:25 +07:00
anxiang	882125e8a2	updated check NaN in batch	2021-08-08 23:36:30 +08:00
AnXiang	cfa54d80c7	updated README.md	2021-08-08 21:34:00 +08:00
anxiangsir	470dfd6842	updated run.sh	2021-07-23 15:53:39 +08:00
anxiangsir	15be549a8b	updated README.md	2021-07-23 15:41:07 +08:00
anxiangsir	bc7480f194	fixed bugs in torch1.6	2021-07-23 15:27:39 +08:00
anxiangsir	6a8e3bc917	fixed torch1.9	2021-07-23 15:27:19 +08:00
anxiangsir	c0b95cc66f	added speed benchmark	2021-07-22 14:29:50 +08:00
anxiang	bdaee1073b	added docs	2021-07-15 00:02:01 +08:00
anxiang	a7605c7e03	added docs	2021-07-14 23:36:14 +08:00
AnXiang	bdb0623efa	updated docs	2021-07-14 12:20:07 +08:00
AnXiang	39a2287bb5	updated docs	2021-07-14 12:01:57 +08:00
AnXiang	7836ea1094	fixed	2021-07-14 12:01:48 +08:00
AnXiang	f3c2fa5302	fixed warmup	2021-07-14 12:01:21 +08:00
AnXiang	6959f9ab8a	added configs	2021-07-14 12:00:55 +08:00
AnXiang	66739ca6b8	updated partial fc doc string.	2021-07-13 16:13:27 +08:00
AnXiang	785f105e19	updated resize	2021-07-12 14:34:22 +08:00
AnXiang	90d2546fdf	fixed bugs	2021-07-12 14:32:10 +08:00
AnXiang	6dc1faaaa5	updated ICCV2021-MFR	2021-07-08 18:47:39 +08:00
AnXiang	f089b66041	updated ICCV2021-MFR	2021-07-08 18:07:32 +08:00
AnXiang	cc488b94a6	updated config examples	2021-07-07 19:22:55 +08:00
AnXiang	3d42efa397	updated code format	2021-07-07 18:35:27 +08:00
AnXiang	95b10ed2ca	updated configs	2021-07-07 18:35:13 +08:00
AnXiang	996d844e12	Update README.md	2021-06-28 20:37:33 +08:00
AnXiang	e1300d3277	updated training resnet2060 command	2021-06-23 13:55:18 +08:00
AnXiang	f7007b6729	Resnet2060! ... updated gradient checkpoint for training resnet2060	2021-06-23 13:34:48 +08:00
AnXiang	676b02d8e6	updated code format	2021-06-23 13:32:35 +08:00
anxiang	a2a3d74e09	added resnet2060 training log	2021-06-23 04:00:08 +08:00
AnXiang	fb5b42fe28	Merge pull request #1546 from ShiquanYu/feature/mobilefacenet ... Add mobilefacenet for arcface_torch	2021-06-07 10:30:14 +08:00
Jia Guo	b5591686f1	Update README.md	2021-05-31 16:10:42 +08:00
shiquanyu	bb611ed662	add mobilefacenet	2021-05-31 09:50:35 +08:00
shiquanyu	33ca5ebc6b	fixed error while change embedding size	2021-05-31 09:50:06 +08:00
007gzs	b129ae9cb3	add simplifier in torch2onnx	2021-05-19 18:47:30 +08:00
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