11 Commits

Author	SHA1	Message	Date
liaoxingyu	ac8409a7da	updating for pytorch1.6	2020-08-20 15:51:41 +08:00
liaoxingyu	16655448c2	onnx/trt support ... Summary: change model pretrain mode and support onnx/TensorRT export	2020-07-29 17:43:39 +08:00
liaoxingyu	fec7abc461	finish v0.2 ddp training	2020-07-06 16:57:43 +08:00
liaoxingyu	4cc157a12f	add lars optimizer ... Summary: support lars optimizer refer to https://arxiv.org/abs/1708.03888 which can make large batch training stable	2020-06-22 12:12:19 +08:00
liaoxingyu	0b15ac4e03	feat(hooks&optim): update stochastic weight averging hooks ... Update swa method which will do after regular training if you set this option enabled.	2020-05-08 12:20:04 +08:00
liaoxingyu	948af64fd1	feat: add swa algorithm ... Add swa and related config options, if it is enabled, model will do swa after regular training	2020-05-06 10:17:44 +08:00
liaoxingyu	325d9abb76	feat($solver): change scheduler call methods ... using name of lr scheduler in config to call	2020-04-27 15:12:01 +08:00
liaoxingyu	95a3c62ad2	refactor(fastreid) ... refactor architecture	2020-04-20 10:59:29 +08:00
liaoxingyu	be9faa5605	update focal loss ... update dataset info display update seperate lr update adaptive label smooth regularization	2020-04-17 13:46:10 +08:00
liaoxingyu	23bedfce12	update version0.2 code	2020-03-25 10:58:26 +08:00
L1aoXingyu	db6ed12b14	Update sampler code	2020-02-10 07:38:56 +08:00