6 Commits (a327a70f0d27a7ebff9f355bf1d6faa5756b127a)

Author	SHA1	Message	Date
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
liaoxingyu2	a00e50d37f	fix triplet ddp training ... Summary: fixup precision alignment when triplet loss with ddp	2020-11-06 11:01:10 +08:00
liaoxingyu	3d1bae9f13	fix triplet loss backward propagation on multi-gpu training (#82) ... Summary: fix torch.distributed.all_gather has no gradient when performing all_gather operation on the provided tensors, instead using `GatherLayer`.	2020-09-28 17:16:51 +08:00
liaoxingyu	d00ce8fc3c	refactor model arch	2020-09-01 16:14:45 +08:00
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