mmpretrain/tests/test_models/test_backbones
Hakjin Lee 05e4bc17b2
[Feature] Support Activation Checkpointing for ConvNeXt. (#1152)
* Support Activation Checkpointing for ConvNeXt

* Add test case

* Lint

* Add docstring
2022-11-14 17:08:31 +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 [Feature] Support Activation Checkpointing for ConvNeXt. (#1152) 2022-11-14 17:08:31 +08:00
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_efficientformer.py [Feature] Support HorNet Backbone. (#1013) 2022-09-27 10:37:49 +08:00
test_efficientnet.py [Enhance] Upgrade isort pre-commit hooks. (#687) 2022-02-17 02:17:20 +08:00
test_hornet.py [Feature] Support HorNet Backbone. (#1013) 2022-09-27 10:37:49 +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 [Enhance] Improve downstream repositories compatibility (#421) 2021-09-08 10:38:57 +08:00
test_mobilenet_v3.py [Enhance] Improve downstream repositories compatibility (#421) 2021-09-08 10:38:57 +08:00
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 [Enhance] Improve downstream repositories compatibility (#421) 2021-09-08 10:38:57 +08:00
test_repmlp.py [Feature] ADD RepMLP Backbone and checkpoints. (#709) 2022-03-30 20:40:25 +08:00
test_repvgg.py [Enhance] RepVGG for YOLOX-PAI. (#1025) 2022-09-30 14:20:53 +08:00
test_res2net.py [Feature] Add Res2Net backbone and converted weights. (#465) 2021-10-20 16:34:22 +08:00
test_resnest.py [Docs] Add Copyright information. (#413) 2021-08-17 19:52:42 +08:00
test_resnet.py [Feature] Add ResNetV1c. (#692) 2022-02-23 11:36:33 +08:00
test_resnet_cifar.py [Docs] Add Copyright information. (#413) 2021-08-17 19:52:42 +08:00
test_resnext.py [Enhance] Improve downstream repositories compatibility (#421) 2021-09-08 10:38:57 +08:00
test_seresnet.py [Enhance] Improve downstream repositories compatibility (#421) 2021-09-08 10:38:57 +08:00
test_seresnext.py [Enhance] Improve downstream repositories compatibility (#421) 2021-09-08 10:38:57 +08:00
test_shufflenet_v1.py [Enhance] Improve downstream repositories compatibility (#421) 2021-09-08 10:38:57 +08:00
test_shufflenet_v2.py [Enhance] Improve downstream repositories compatibility (#421) 2021-09-08 10:38:57 +08:00
test_swin_transformer.py [Fix] Fix the output position of Swin-Transformer. (#947) 2022-08-03 19:32:29 +08:00
test_swin_transformer_v2.py [Feature] Support Swin Transform V2. (#799) 2022-08-03 17:33:08 +08:00
test_t2t_vit.py [Refactor] Re-write get_sinusoid_encoding from third-party implementation. (#965) 2022-09-13 15:24:29 +08:00
test_timm_backbone.py [Feature] Support MPS device. (#894) 2022-07-28 12:28:51 +08:00
test_tnt.py [Refactor] Fix TnT compatibility and verbose warning. (#436) 2021-09-22 19:37:24 +08:00
test_twins.py [Feature] Add Twins backbone and convert checkpoints. (#642) 2022-01-27 10:25:05 +08:00
test_van.py [Feature] Support VAN. (#739) 2022-04-28 09:35:17 +08:00
test_vgg.py [Enhance] Improve downstream repositories compatibility (#421) 2021-09-08 10:38:57 +08:00
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