mirrors
/
mmpretrain
mirror of https://github.com/open-mmlab/mmpretrain.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
mmpretrain/mmcls/models/backbones
History
Ma Zerun c708770b42
[Enhance] Support dynamic input shape for ViT-based algorithms. (#706) 
* Move `resize_pos_embed` to `mmcls.models.utils`

* Refactor Vision Transformer

* Refactor DeiT

* Refactor MLP-Mixer

* Refactor Swin-Transformer

* Remove `indexing` arg

* Support dynamic inputs for t2t_vit

* Add copyright

* Fix bugs in swin transformer

* Add `pad_small_maps` option

* Update swin transformer

* Handle `attn_mask` in checkpoints of swin

* Imporve by comments
 		2022-03-03 13:10:12 +08:00
..
__init__.py [Feature] Add ResNetV1c. (#692) 2022-02-23 11:36:33 +08:00
alexnet.py [Enhance] Improve downstream repositories compatibility (#421) 2021-09-08 10:38:57 +08:00
base_backbone.py [Docs] Add Copyright information. (#413) 2021-08-17 19:52:42 +08:00
conformer.py [Fix] Fix Conformer forward with irregular input size. (#686) 2022-02-28 10:17:52 +08:00
convnext.py [Feature] Support ConvNeXt (#670) 2022-01-28 10:36:45 +08:00
deit.py [Enhance] Support dynamic input shape for ViT-based algorithms. (#706) 2022-03-03 13:10:12 +08:00
efficientnet.py [Feature] Support EfficientNet (#649) 2022-01-25 12:14:17 +08:00
hrnet.py [Feature] Support HRNet and add pre-trained models (#660) 2022-01-28 10:54:14 +08:00
lenet.py [Enhance] Improve downstream repositories compatibility (#421) 2021-09-08 10:38:57 +08:00
mlp_mixer.py [Enhance] Support dynamic input shape for ViT-based algorithms. (#706) 2022-03-03 13:10:12 +08:00
mobilenet_v2.py [Enhance] Improve downstream repositories compatibility (#421) 2021-09-08 10:38:57 +08:00
mobilenet_v3.py [Docs] Add code-spell pre-commit hook and fix a large mount of typos. (#470) 2021-10-13 14:33:07 +08:00
regnet.py [Fix] Fix missing import `Compose` and `Normalize`. 2021-10-28 15:21:05 +08:00
repvgg.py [Enhance] Upgrade isort pre-commit hooks. (#687) 2022-02-17 02:17:20 +08:00
res2net.py [Feature] Add Res2Net backbone and converted weights. (#465) 2021-10-20 16:34:22 +08:00
resnest.py [Docs] Add Copyright information. (#413) 2021-08-17 19:52:42 +08:00
resnet.py [Feature] Add ResNetV1c. (#692) 2022-02-23 11:36:33 +08:00
resnet_cifar.py [Enhance] Improve downstream repositories compatibility (#421) 2021-09-08 10:38:57 +08:00
resnext.py [Docs] Add Copyright information. (#413) 2021-08-17 19:52:42 +08:00
seresnet.py [Docs] Add Copyright information. (#413) 2021-08-17 19:52:42 +08:00
seresnext.py [Docs] Add Copyright information. (#413) 2021-08-17 19:52:42 +08:00
shufflenet_v1.py [Enhance] Improve downstream repositories compatibility (#421) 2021-09-08 10:38:57 +08:00
shufflenet_v2.py [Enhance] Rewrite channel split operation in ShufflenetV2 (#632) 2022-01-25 14:45:28 +08:00
swin_transformer.py [Enhance] Support dynamic input shape for ViT-based algorithms. (#706) 2022-03-03 13:10:12 +08:00
t2t_vit.py [Enhance] Support dynamic input shape for ViT-based algorithms. (#706) 2022-03-03 13:10:12 +08:00
timm_backbone.py [Feature] Support features_only in TIMMBackbone (#668) 2022-01-25 11:22:23 +08:00
tnt.py [Docs] Add code-spell pre-commit hook and fix a large mount of typos. (#470) 2021-10-13 14:33:07 +08:00
twins.py [Feature] Add Twins backbone and convert checkpoints. (#642) 2022-01-27 10:25:05 +08:00
vgg.py [Docs] Add code-spell pre-commit hook and fix a large mount of typos. (#470) 2021-10-13 14:33:07 +08:00
vision_transformer.py [Enhance] Support dynamic input shape for ViT-based algorithms. (#706) 2022-03-03 13:10:12 +08:00