liaoxingyu
|
16655448c2
|
onnx/trt support
Summary: change model pretrain mode and support onnx/TensorRT export
|
2020-07-29 17:43:39 +08:00 |
|
liaoxingyu
|
3b57dea49f
|
support regnet backbone
|
2020-07-17 19:13:45 +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
|
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
|
0356ef8c5c
|
feat: add SyncBN and GroupNorm suppor
|
2020-05-14 11:36:28 +08:00 |
|
liaoxingyu
|
8abd3bab03
|
feat($layers): add new act func
add mish, gelu supported
|
2020-04-24 12:17:00 +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
|
9cf222e093
|
refactor bn_no_bias
|
2020-04-08 21:04:09 +08:00 |
|
liaoxingyu
|
23bedfce12
|
update version0.2 code
|
2020-03-25 10:58:26 +08:00 |
|
L1aoXingyu
|
8a9c0ccfad
|
Finish first version for fastreid
|
2020-02-10 22:13:04 +08:00 |
|
L1aoXingyu
|
db6ed12b14
|
Update sampler code
|
2020-02-10 07:38:56 +08:00 |
|
liaoxingyu
|
b761b656f3
|
Finish basic training loop and evaluation results
|
2020-01-20 21:33:37 +08:00 |
|