mmclassification/tests/test_models/test_backbones
Ma Zerun 5de480ea9e
[Feature] Support HRNet and add pre-trained models (#660)
* Support HRNet

* Add HRNet configs

* Fix a bug in backward

* Add configs and update docs.

* Not use bias in conv before batch norm

* Defaults to use `norm_eval=False`

* Add unit tests and support out_channels in HRFuseScales

* Update checkpoint path

* Update docstring.

* Remove incorrect files

* Improve according to comments
2022-01-28 10:54:14 +08:00
..
test_conformer.py [Feature] Implement the conformer backbone. (#494) 2021-12-07 14:00:17 +08:00
test_convnext.py [Feature] Support ConvNeXt (#670) 2022-01-28 10:36:45 +08:00
test_deit.py [Feature] Add DeiT backbone and checkpoints. (#576) 2021-12-15 22:44:57 +08:00
test_efficientnet.py [Feature] Support EfficientNet (#649) 2022-01-25 12:14:17 +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 [Feature] Add MLP Mixer Backbone. (#528) 2021-11-15 18:25:22 +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_regnet.py [Enhance] Improve downstream repositories compatibility (#421) 2021-09-08 10:38:57 +08:00
test_repvgg.py [Docs] Add code-spell pre-commit hook and fix a large mount of typos. (#470) 2021-10-13 14:33:07 +08:00
test_res2net.py [Feature] Add Res2Net backbone and converted weights. (#465) 2021-10-20 16:34:22 +08:00
test_resnest.py
test_resnet.py [Enhancement] Improve the accuracy of ResNet (#572) 2021-12-29 10:58:42 +08:00
test_resnet_cifar.py
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 [Feature] Add the frozen function for Swin Transformer model. (#574) 2021-12-07 11:58:14 +08:00
test_t2t_vit.py [Feature] Add Tokens-to-Token ViT backbone and converted checkpoints. (#467) 2021-10-29 10:37:16 +08:00
test_timm_backbone.py [Feature] Support features_only in TIMMBackbone (#668) 2022-01-25 11:22:23 +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_vgg.py [Enhance] Improve downstream repositories compatibility (#421) 2021-09-08 10:38:57 +08:00
test_vision_transformer.py [Refactor] Refator ViT (Continue #295) (#395) 2021-10-18 16:07:00 +08:00