Ma Zerun 556fa567a8

[Feature] Support MViT and add checkpoints. (#924) ...

* [Feature] Support MViT.

* Add MViT configs and docs

* Add unit test

* Fix unit tests.

2022-08-02 15:20:16 +08:00

..

__init__.py

[Enhance] Support dynamic input shape for ViT-based algorithms. (#706)

2022-03-03 13:10:12 +08:00

test_conformer.py

…

test_convmixer.py

[Feature] Support ConvMixer. (#716)

2022-03-23 15:23:57 +08:00

test_convnext.py

…

test_cspnet.py

[Feature] Add CSPNet and backbone and checkpoints (#735)

2022-03-30 15:29:35 +08:00

test_deit.py

[Enhance] Support dynamic input shape for ViT-based algorithms. (#706)

2022-03-03 13:10:12 +08:00

test_densenet.py

[Feature] Support DenseNet. (#750)

2022-04-28 13:06:03 +08:00

test_efficientnet.py

…

test_hrnet.py

…

test_mlp_mixer.py

[Enhance] Support dynamic input shape for ViT-based algorithms. (#706)

2022-03-03 13:10:12 +08:00

test_mobilenet_v2.py

…

test_mobilenet_v3.py

…

test_mvit.py

[Feature] Support MViT and add checkpoints. (#924)

2022-08-02 15:20:16 +08:00

test_poolformer.py

[Feature] Add PoolFormer backbone and checkpoints. (#746)

2022-04-15 11:35:15 +08:00

test_regnet.py

…

test_repmlp.py

[Feature] ADD RepMLP Backbone and checkpoints. (#709)

2022-03-30 20:40:25 +08:00

test_repvgg.py

…

test_res2net.py

…

test_resnest.py

…

test_resnet_cifar.py

…

test_resnet.py

…

test_resnext.py

…

test_seresnet.py

…

test_seresnext.py

…

test_shufflenet_v1.py

…

test_shufflenet_v2.py

…

test_swin_transformer.py

[Enhance] Reduce the memory usage of unit tests for Swin-Transformer. (#759)

2022-04-22 17:41:15 +08:00

test_t2t_vit.py

[Fix] Reduce unit test memory usage of T2T-ViT

2022-05-16 17:01:30 +08:00

test_timm_backbone.py

[Feature] Support MPS device. (#894)

2022-07-28 12:28:51 +08:00

test_tnt.py

…

test_twins.py

…

test_van.py

[Feature] Support VAN. (#739)

2022-04-28 09:35:17 +08:00

test_vgg.py

…

test_vision_transformer.py

[Enhance] Support dynamic input shape for ViT-based algorithms. (#706)

2022-03-03 13:10:12 +08:00

utils.py

[Enhance] Support dynamic input shape for ViT-based algorithms. (#706)

2022-03-03 13:10:12 +08:00