mmpretrain/configs/_base_/models
Ma Zerun dbf3df21a3
[Refactor] Use `out_type` to specify ViT-like backbone output. (#1408)
* [Refactor] Use  to specify ViT-like backbone output.

* Fix ClsBatchNormNeck

* Update mmpretrain/models/necks/mae_neck.py

---------

Co-authored-by: Yixiao Fang <36138628+fangyixiao18@users.noreply.github.com>
2023-03-09 11:02:58 +08:00
..
conformer [Improve] Speed up data preprocessor. (#1064) 2022-10-17 17:08:18 +08:00
convmixer [Feature] Support ConvMixer. (#716) 2022-03-23 15:23:57 +08:00
convnext [Feature] Support ConvNeXt-V2 backbone. (#1294) 2023-01-06 16:13:41 +08:00
convnext_v2 [Feature] Support ConvNeXt-V2 backbone. (#1294) 2023-01-06 16:13:41 +08:00
davit [Feature] Support DaViT. (#1105) 2022-11-16 17:23:55 +08:00
deit3 [Fix] Fix configs due to api refactor of `num_classes`. (#1184) 2022-11-14 14:43:56 +08:00
densenet [Feature] Support DenseNet. (#750) 2022-04-28 13:06:03 +08:00
edgenext [Feature] Support EdgeNeXt for dev-1.x. (#1037) 2022-09-21 13:27:04 +08:00
efficientnet_v2 [Improve] Remove useless EfficientnetV2 config files. (#1300) 2023-01-06 16:10:59 +08:00
eva [Refactor] Use `out_type` to specify ViT-like backbone output. (#1408) 2023-03-09 11:02:58 +08:00
hornet [Feature] Support HorNet Backbone for dev1.x. (#1094) 2022-11-04 15:33:46 +08:00
hrnet [Feature] Support HRNet and add pre-trained models (#660) 2022-01-28 10:54:14 +08:00
mixmim [Feature] Add mixmim backbone with checkpoints. (#1224) 2022-12-20 16:52:54 +08:00
mobilenet_v3 [Enhance] Add more mobilenetv3 pretrains. (#1154) 2022-11-14 17:06:39 +08:00
mobileone [Feature] Add MobileOne Backbone For MMCls 1.x. (#1030) 2022-09-16 11:55:10 +08:00
mobilevit [Feature] Support MobileViT backbone. (#1068) 2022-10-18 17:05:59 +08:00
mvit [Fix] Fix configs due to api refactor of `num_classes`. (#1184) 2022-11-14 14:43:56 +08:00
poolformer [Feature] Add PoolFormer backbone and checkpoints. (#746) 2022-04-15 11:35:15 +08:00
regnet Add RegNetX configs (#269) 2021-06-04 14:29:13 +08:00
revvit [Refactor] Use `out_type` to specify ViT-like backbone output. (#1408) 2023-03-09 11:02:58 +08:00
swin_transformer [Improve] Speed up data preprocessor. (#1064) 2022-10-17 17:08:18 +08:00
swin_transformer_v2 [Docs] Improve Swin Doc and Add Tabs enxtation. (#1145) 2022-11-04 11:58:39 +08:00
tinyvit [Feature] Add TinyViT for dev-1.x. (#1042) 2022-12-20 13:04:00 +08:00
van [Improve] Speed up data preprocessor. (#1064) 2022-10-17 17:08:18 +08:00
vig [Feature] Support VIG Backbone. (#1304) 2023-01-17 16:55:56 +08:00
efficientformer-l1.py [Feature] Add efficientformer Backbone for MMCls 1.x. (#1031) 2022-09-20 14:56:45 +08:00
efficientnet_b0.py [Feature] Support EfficientNet (#649) 2022-01-25 12:14:17 +08:00
efficientnet_b1.py [Feature] Support EfficientNet (#649) 2022-01-25 12:14:17 +08:00
efficientnet_b2.py [Feature] Support EfficientNet (#649) 2022-01-25 12:14:17 +08:00
efficientnet_b3.py [Feature] Support EfficientNet (#649) 2022-01-25 12:14:17 +08:00
efficientnet_b4.py [Feature] Support EfficientNet (#649) 2022-01-25 12:14:17 +08:00
efficientnet_b5.py [Feature] Support EfficientNet (#649) 2022-01-25 12:14:17 +08:00
efficientnet_b6.py [Feature] Support EfficientNet (#649) 2022-01-25 12:14:17 +08:00
efficientnet_b7.py [Feature] Support EfficientNet (#649) 2022-01-25 12:14:17 +08:00
efficientnet_b8.py [Feature] Support EfficientNet (#649) 2022-01-25 12:14:17 +08:00
efficientnet_em.py [Feature] Support EfficientNet (#649) 2022-01-25 12:14:17 +08:00
efficientnet_es.py [Feature] Support EfficientNet (#649) 2022-01-25 12:14:17 +08:00
efficientnet_l2.py [Feature] EfficientNets NoisyStudent & L2. (#1122) 2022-11-21 16:57:13 +08:00
inception_v3.py [Refactor] Refactor dev scripts 2022-07-18 11:11:13 +08:00
levit-256-p16.py [Feature] Support LeViT backbone. (#1238) 2023-01-17 17:43:42 +08:00
mae_vit-base-p16.py [Refactor] Add necks, heads and losses for the self-supervised task. (#1376) 2023-02-28 10:05:00 +08:00
mlp_mixer_base_patch16.py [Feature] Add MLP Mixer Backbone. (#528) 2021-11-15 18:25:22 +08:00
mlp_mixer_large_patch16.py [Enhance] Add metafile, readme and converted models for MLP-Mixer (#539) 2021-11-24 19:04:19 +08:00
mobilenet_v2_1x.py Add configs for MNIST, CIFAR10 and ImageNet 2020-07-12 00:06:57 +08:00
replknet-31B_in1k.py Rename the package name to `mmpretrain`. 2023-02-17 15:20:55 +08:00
replknet-31L_in1k.py [Feature] Support RepLKnet backbone. (#1129) 2022-11-21 10:18:58 +08:00
replknet-XL_in1k.py [Feature] Support RepLKnet backbone. (#1129) 2022-11-21 10:18:58 +08:00
repmlp-base_224.py [Feature] ADD RepMLP Backbone and checkpoints. (#709) 2022-03-30 20:40:25 +08:00
repvgg-A0_in1k.py [Feature] Add RepVGG backbone and checkpoints. (#414) 2021-09-29 11:06:23 +08:00
repvgg-B3_lbs-mixup_in1k.py [Improve] Speed up data preprocessor. (#1064) 2022-10-17 17:08:18 +08:00
res2net50-w14-s8.py [Feature] Add Res2Net backbone and converted weights. (#465) 2021-10-20 16:34:22 +08:00
res2net50-w26-s4.py [Feature] Add Res2Net backbone and converted weights. (#465) 2021-10-20 16:34:22 +08:00
res2net50-w26-s6.py [Feature] Add Res2Net backbone and converted weights. (#465) 2021-10-20 16:34:22 +08:00
res2net50-w26-s8.py [Feature] Add Res2Net backbone and converted weights. (#465) 2021-10-20 16:34:22 +08:00
res2net50-w48-s2.py [Feature] Add Res2Net backbone and converted weights. (#465) 2021-10-20 16:34:22 +08:00
res2net101-w26-s4.py [Feature] Add Res2Net backbone and converted weights. (#465) 2021-10-20 16:34:22 +08:00
resnest50.py [Improve] Speed up data preprocessor. (#1064) 2022-10-17 17:08:18 +08:00
resnest101.py [Improve] Speed up data preprocessor. (#1064) 2022-10-17 17:08:18 +08:00
resnest200.py [Improve] Speed up data preprocessor. (#1064) 2022-10-17 17:08:18 +08:00
resnest269.py [Improve] Speed up data preprocessor. (#1064) 2022-10-17 17:08:18 +08:00
resnet18.py Add configs for MNIST, CIFAR10 and ImageNet 2020-07-12 00:06:57 +08:00
resnet18_cifar.py Add configs for MNIST, CIFAR10 and ImageNet 2020-07-12 00:06:57 +08:00
resnet34.py Add configs for MNIST, CIFAR10 and ImageNet 2020-07-12 00:06:57 +08:00
resnet34_cifar.py Add CIFAR10 configs and models (#38) 2020-08-26 16:34:15 +08:00
resnet34_gem.py [Feature] Support gem pooling (#677) 2022-02-17 02:17:36 +08:00
resnet50.py Add configs for MNIST, CIFAR10 and ImageNet 2020-07-12 00:06:57 +08:00
resnet50_cifar.py Add configs for MNIST, CIFAR10 and ImageNet 2020-07-12 00:06:57 +08:00
resnet50_cifar_cutmix.py [Feature]Add augments to models/utils (#278) 2021-06-20 09:44:51 +08:00
resnet50_cifar_mixup.py [Improve] Speed up data preprocessor. (#1064) 2022-10-17 17:08:18 +08:00
resnet50_cutmix.py [Feature]Add augments to models/utils (#278) 2021-06-20 09:44:51 +08:00
resnet50_label_smooth.py [Bug]Fix label smooth bug (#203) 2021-04-13 13:53:56 +08:00
resnet50_mixup.py [Improve] Speed up data preprocessor. (#1064) 2022-10-17 17:08:18 +08:00
resnet101.py Add configs for MNIST, CIFAR10 and ImageNet 2020-07-12 00:06:57 +08:00
resnet101_cifar.py Add CIFAR10 configs and models (#38) 2020-08-26 16:34:15 +08:00
resnet152.py Add configs for MNIST, CIFAR10 and ImageNet 2020-07-12 00:06:57 +08:00
resnet152_cifar.py Add CIFAR10 configs and models (#38) 2020-08-26 16:34:15 +08:00
resnetv1c50.py [Feature] Add ResNetV1c. (#692) 2022-02-23 11:36:33 +08:00
resnetv1d50.py Add configs for MNIST, CIFAR10 and ImageNet 2020-07-12 00:06:57 +08:00
resnetv1d101.py Add configs for MNIST, CIFAR10 and ImageNet 2020-07-12 00:06:57 +08:00
resnetv1d152.py Add configs for MNIST, CIFAR10 and ImageNet 2020-07-12 00:06:57 +08:00
resnext50_32x4d.py Add configs for MNIST, CIFAR10 and ImageNet 2020-07-12 00:06:57 +08:00
resnext101_32x4d.py Add configs for MNIST, CIFAR10 and ImageNet 2020-07-12 00:06:57 +08:00
resnext101_32x8d.py Add configs for MNIST, CIFAR10 and ImageNet 2020-07-12 00:06:57 +08:00
resnext152_32x4d.py Add configs for MNIST, CIFAR10 and ImageNet 2020-07-12 00:06:57 +08:00
seresnet50.py Add configs for MNIST, CIFAR10 and ImageNet 2020-07-12 00:06:57 +08:00
seresnet101.py Add configs for MNIST, CIFAR10 and ImageNet 2020-07-12 00:06:57 +08:00
seresnext50_32x4d.py Add configs for MNIST, CIFAR10 and ImageNet 2020-07-12 00:06:57 +08:00
seresnext101_32x4d.py Add configs for MNIST, CIFAR10 and ImageNet 2020-07-12 00:06:57 +08:00
shufflenet_v1_1x.py Update shufflenet config (#31) 2020-08-13 10:46:39 +08:00
shufflenet_v2_1x.py Update shufflenet config (#31) 2020-08-13 10:46:39 +08:00
t2t-vit-t-14.py [Improve] Speed up data preprocessor. (#1064) 2022-10-17 17:08:18 +08:00
t2t-vit-t-19.py [Improve] Speed up data preprocessor. (#1064) 2022-10-17 17:08:18 +08:00
t2t-vit-t-24.py [Improve] Speed up data preprocessor. (#1064) 2022-10-17 17:08:18 +08:00
tnt_s_patch16_224.py [Feature] Add transformer in transformer (#339) 2021-07-30 09:16:27 +08:00
twins_pcpvt_base.py [Improve] Speed up data preprocessor. (#1064) 2022-10-17 17:08:18 +08:00
twins_svt_base.py [Improve] Speed up data preprocessor. (#1064) 2022-10-17 17:08:18 +08:00
vgg11.py Add VGG and pretained models (#27) 2020-09-29 17:49:42 +08:00
vgg11bn.py Add VGG and pretained models (#27) 2020-09-29 17:49:42 +08:00
vgg13.py Add VGG and pretained models (#27) 2020-09-29 17:49:42 +08:00
vgg13bn.py Add VGG and pretained models (#27) 2020-09-29 17:49:42 +08:00
vgg16.py Add VGG and pretained models (#27) 2020-09-29 17:49:42 +08:00
vgg16bn.py Add VGG and pretained models (#27) 2020-09-29 17:49:42 +08:00
vgg19.py Add VGG and pretained models (#27) 2020-09-29 17:49:42 +08:00
vgg19bn.py Add VGG and pretained models (#27) 2020-09-29 17:49:42 +08:00
vit-base-p16.py [Refactor] Refator ViT (Continue #295) (#395) 2021-10-18 16:07:00 +08:00
vit-base-p32.py [Refactor] Refator ViT (Continue #295) (#395) 2021-10-18 16:07:00 +08:00
vit-large-p16.py [Refactor] Refator ViT (Continue #295) (#395) 2021-10-18 16:07:00 +08:00
vit-large-p32.py [Refactor] Refator ViT (Continue #295) (#395) 2021-10-18 16:07:00 +08:00
wide-resnet50.py [Feature] Support Wide-ResNet. (#715) 2022-03-04 14:06:34 +08:00