14 Commits (7f747235baf559f69438856da687abc1f99f277b)

Author	SHA1	Message	Date
Xingyu Liao	15c556c43a	remove apex dependency (#442) ... Summary: Use Pytorch1.6(or above) built-in amp training	2021-03-23 12:12:35 +08:00
liaoxingyu	ef6ebf451b	refactor apex import	2021-01-23 15:35:48 +08:00
liaoxingyu	e26182e6ec	make lr warmup by iter ... Summary: change warmup way by iter not by epoch, which will make it more flexible when training small epochs	2021-01-22 11:17:21 +08:00
liaoxingyu	15e1729a27	update fastreid V1.0	2021-01-18 11:36:38 +08:00
liaoxingyu	0e1b91f74a	fix checkpoint bug in ddp training ... Summary: change pytorch ddp to apex ddp in checkpoint	2020-12-28 14:35:02 +08:00
liaoxingyu	bb7a00e615	feat: add save best model checkpoint	2020-12-22 15:50:23 +08:00
liaoxingyu	a327a70f0d	v0.3 update ... Summary: 1. change DPP training in apex way; 2. make warmup scheduler by iter and lr scheduler by epoch; 3. replace random erasing with torchvision implementation; 4. naming modification in config file	2020-12-07 14:19:20 +08:00
liaoxingyu	7e9a4775da	fixup finetune problem ... Summary: support finetune from the other model with different number of classes, and simplify calling way (#325) close #325 close #325	2020-11-06 15:58:22 +08:00
liaoxingyu	fec7abc461	finish v0.2 ddp training	2020-07-06 16:57:43 +08:00
liaoxingyu	8879db3fba	update training instruction ... Summary: update dataset configuration and training instruction	2020-06-16 19:43:36 +08:00
liaoxingyu	a2dcd7b4ab	feat(layers/norm): add ghost batchnorm ... add a get_norm fucntion to easily change normalization between batchnorm, ghost bn and group bn	2020-05-01 09:02:46 +08:00
liaoxingyu	9cf222e093	refactor bn_no_bias	2020-04-08 21:04:09 +08:00
liaoxingyu	6a8961ce48	1. upload circle loss and arcface ... 2. finish freeze training 3. update augmix data augmentation	2020-04-05 23:54:26 +08:00
L1aoXingyu	db6ed12b14	Update sampler code	2020-02-10 07:38:56 +08:00