9 Commits

Author SHA1 Message Date
liaoxingyu
2cabc3428a Support vision transformer backbone 2021-05-31 17:08:57 +08:00
Xingyu Liao
9b5af4166e
support repvgg (#429)
Summary:
* support repvgg backbone, and verify the consistency of train mode and eval mode
* onnx export logger style modification
2021-03-11 14:23:24 +08:00
liaoxingyu
b786001ebd add mobilenetv2 support 2021-01-26 18:17:01 +08:00
liaoxingyu
15e1729a27 update fastreid V1.0 2021-01-18 11:36:38 +08:00
liaoxingyu
648198e6e5 add efficientnet support 2020-09-10 11:04:52 +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
5ad22d5d36 update regnet init 2020-07-17 19:35:40 +08:00
liaoxingyu
e3ae03cc58 feat($modeling/backbones): add new backbones
add osnet, resnext and resnest backbone supported
2020-04-24 12:14:56 +08:00
L1aoXingyu
db6ed12b14 Update sampler code 2020-02-10 07:38:56 +08:00