liaoxingyu
a53fd17874
update docs
2021-01-23 15:25:58 +08:00
liaoxingyu
b5c3c0a24d
update docs
2021-01-22 21:11:19 +08:00
liaoxingyu
766f309d03
feat: update pairwise cosface and pairwise circle loss
2020-12-22 15:51:49 +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
f496193f17
change cross_entroy_loss input name
...
Summary: change `pred_class_logits` to `pred_class_outputs` to prevent misleading. (#318 )
close #318
2020-11-06 14:16:31 +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
df823da09a
fix typro bce loss
...
close #273
2020-09-18 10:39:10 +08:00
liaoxingyu
d9a63a959f
fix bug in mgn #272
...
Summary: fix get_norm bug in mgn
2020-09-17 18:17:53 +08:00
liaoxingyu
aa5c422606
fix pair-wise circle loss
...
fix #252
2020-09-09 15:28:52 +08:00
liaoxingyu
53fed7451d
feat: support amp training
...
Summary: support automatic mixed precision training #217
2020-09-02 18:03:12 +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
35794621cc
remove addmm warning
2020-07-31 16:32:10 +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
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
ecc2b1a790
update naive sampler
...
Summary: update naive sampler which will introduce unbalanced sampling
2020-06-15 20:50:25 +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
cbdc01a1c3
update pairwise circle loss
...
Summary: add param of pairwise circle loss to config, and update pairwise circle loss version
2020-06-10 19:07:29 +08:00
liaoxingyu
d4b71de3aa
switch between soft and hard margin when inf
...
Summary: Add a mechnism to automatic switch triplet loss with soft margin to hard margin when loss becomes inf.
2020-05-26 14:36:33 +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
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
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
4d2fa28dbb
update freeze layer
...
update preciseBN
update circle loss with metric learning and cross entropy loss form
update loss call methods
2020-04-06 23:34:27 +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
a2f69d0537
Update StrongBaseline results for market1501 and dukemtmc
2020-02-11 22:38:40 +08:00
L1aoXingyu
8a9c0ccfad
Finish first version for fastreid
2020-02-10 22:13:04 +08:00