mmpretrain/mmcls/models/backbones
fangxu 0aa789f3c3
[Feature] Add the frozen function for Swin Transformer model. (#574)
* Add the frozen function for Swin Transformer model

* add frozen parameter for swin transformer model

* add norm_eval parameter

* Delete =11.1

* Delete =418,driver

* delete _BatchNorm

* remove LayerNorm , add _BatchNorm

* unifying the style of frozen function refer ResNet

* Improve docs and add unit tests.

Co-authored-by: cxiang26 <cq.xiang@foxmail.com>
Co-authored-by: mzr1996 <mzr1996@163.com>
2021-12-07 11:58:14 +08:00
..
__init__.py [Feature] Add MLP Mixer Backbone. (#528) 2021-11-15 18:25:22 +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
lenet.py [Enhance] Improve downstream repositories compatibility (#421) 2021-09-08 10:38:57 +08:00
mlp_mixer.py [Enhance] Add metafile, readme and converted models for MLP-Mixer (#539) 2021-11-24 19:04:19 +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 [Docs] Add code-spell pre-commit hook and fix a large mount of typos. (#470) 2021-10-13 14:33:07 +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 Res2Net backbone and converted weights. (#465) 2021-10-20 16:34:22 +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] Improve downstream repositories compatibility (#421) 2021-09-08 10:38:57 +08:00
swin_transformer.py [Feature] Add the frozen function for Swin Transformer model. (#574) 2021-12-07 11:58:14 +08:00
t2t_vit.py [Docs] Fix docs build dependency. (#584) 2021-12-07 11:16:33 +08:00
timm_backbone.py [Refactor] Use `reset_classifier` to remove head of timm backbones. (#534) 2021-11-16 16:24:55 +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
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 [Docs] Fix docs build dependency. (#584) 2021-12-07 11:16:33 +08:00