mirrors
/
fast-reid
mirror of https://github.com/JDAI-CV/fast-reid.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
458 Commits
1 Branch
4 Tags
15 MiB
Commit Graph

16 Commits (f57c5764e37c9648782c92c55aa7032122bcc30e)

Author SHA1 Message Date
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 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 5b88736e1d refactor evaluation code 2020-09-23 19:32:40 +08:00
liaoxingyu 77caa01e34 fix memory leak in sgd #217 2020-09-10 10:50:22 +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
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