16 Commits

Author SHA1 Message Date
liaoxingyu
d9a63a959f fix bug in mgn #272
Summary: fix get_norm bug in mgn
2020-09-17 18:17:53 +08:00
liaoxingyu
d00ce8fc3c refactor model arch 2020-09-01 16:14:45 +08:00
liaoxingyu
3f35eb449d minor update 2020-07-14 11:58:06 +08:00
liaoxingyu
ea8a3cc534 fix typro 2020-07-10 16:26:35 +08:00
liaoxingyu
fec7abc461 finish v0.2 ddp training 2020-07-06 16:57:43 +08:00
liaoxingyu
56a1ab4a5d update fast global avgpool
Summary: update fast pool according to https://arxiv.org/pdf/2003.13630.pdf
2020-06-12 16:34:03 +08:00
liaoxingyu
bc221cb05f fix mgn multi-gpu training problem
Summary: norm_type in pool_reduce will not change when use syncBN
2020-06-05 11:11:50 +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
84c733fa85 fix: remove prefetcher, put normalizer in model
1. remove messy data prefetcher which will cause  confusion
2. put normliazer in model to accelerate training via GPU computing
2020-05-25 23:39:11 +08:00
liaoxingyu
18a33f7962 feat: add MGN model
support MGN architecture and training config
2020-05-15 11:39:54 +08:00
liaoxingyu
6c9af664dc refactor($modeling/meta_arch): remove useless parts
remove useless meta_archs and backbones
2020-04-21 11:42:14 +08:00
liaoxingyu
95a3c62ad2 refactor(fastreid)
refactor architecture
2020-04-20 10:59:29 +08:00
liaoxingyu
9684500a57 chagne arch
1. change dataset show to trainset show and testset show seperately
2. add cls layer to easily plug in circle loss and arcface
2020-04-19 12:54:01 +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
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