Ma Zerun b5bb86a357

[Fix] Fix the output position of Swin-Transformer. (#947) ...

* [Fix] Fix the output position of Swin-Transformer.

* Rename `downsample` argument to `do_downsample`.

2022-08-03 19:32:29 +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

[Fix] Fix Conformer forward with irregular input size. (#686)

2022-02-28 10:17:52 +08:00

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

[Enhance] Upgrade isort pre-commit hooks. (#687)

2022-02-17 02:17:20 +08:00

test_hrnet.py

[Feature] Support HRNet and add pre-trained models (#660)

2022-01-28 10:54:14 +08:00

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

[Enhance] Upgrade isort pre-commit hooks. (#687)

2022-02-17 02:17:20 +08:00

test_res2net.py

…

test_resnest.py

…

test_resnet_cifar.py

…

test_resnet.py

[Feature] Add ResNetV1c. (#692)

2022-02-23 11:36:33 +08:00

test_resnext.py

…

test_seresnet.py

…

test_seresnext.py

…

test_shufflenet_v1.py

…

test_shufflenet_v2.py

…

test_swin_transformer_v2.py

[Feature] Support Swin Transform V2. (#799)

2022-08-03 17:33:08 +08:00

test_swin_transformer.py

[Fix] Fix the output position of Swin-Transformer. (#947)

2022-08-03 19:32:29 +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