5 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
liaoxingyu	16655448c2	onnx/trt support ... Summary: change model pretrain mode and support onnx/TensorRT export	2020-07-29 17:43:39 +08:00
liaoxingyu	3840f3f79a	fix arcface NaN problem ... Summary: fix classifier init bugs, which will not initialize classifier weights when use arcface or circle loss. In this way, it will lead loss NaN problem.	2020-06-18 12:05:44 +08:00
liaoxingyu	5528d17ace	refactor code ... Summary: change code style and refactor code, add avgmax pooling layer in gem_pool	2020-05-28 13:49:39 +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