mirrors
/
mmpretrain
mirror of https://github.com/open-mmlab/mmpretrain.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
mmpretrain/configs/_base_/models
History
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
..
conformer [Feature] Implement the conformer backbone. (#494) 2021-12-07 14:00:17 +08:00
convnext [Feature] Support ConvNeXt (#670) 2022-01-28 10:36:45 +08:00
hrnet [Feature] Support HRNet and add pre-trained models (#660) 2022-01-28 10:54:14 +08:00
regnet Add RegNetX configs (#269) 2021-06-04 14:29:13 +08:00
swin_transformer [Enhance] Improve downstream repositories compatibility (#421) 2021-09-08 10:38:57 +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
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
mobilenet_v3_large_imagenet.py [Enhance] Enhance feature extraction function. (#593) 2021-12-17 15:55:02 +08:00
mobilenet_v3_small_cifar.py Fix Mobilenetv3 structure and add pretrained model (#291) 2021-06-27 23:19:36 +08:00
mobilenet_v3_small_imagenet.py [Enhance] Enhance feature extraction function. (#593) 2021-12-17 15:55:02 +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 [Feature] Add RepVGG backbone and checkpoints. (#414) 2021-09-29 11:06:23 +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 [Feature] Add ResNeSt configs (#332) 2021-07-14 15:58:19 +08:00
resnest101.py [Feature] Add ResNeSt configs (#332) 2021-07-14 15:58:19 +08:00
resnest200.py [Feature] Add ResNeSt configs (#332) 2021-07-14 15:58:19 +08:00
resnest269.py [Feature] Add ResNeSt configs (#332) 2021-07-14 15:58:19 +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
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 [Feature]Add augments to models/utils (#278) 2021-06-20 09:44:51 +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 [Feature]Add augments to models/utils (#278) 2021-06-20 09:44:51 +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
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 [Feature] Add Tokens-to-Token ViT backbone and converted checkpoints. (#467) 2021-10-29 10:37:16 +08:00
t2t-vit-t-19.py [Feature] Add Tokens-to-Token ViT backbone and converted checkpoints. (#467) 2021-10-29 10:37:16 +08:00
t2t-vit-t-24.py [Feature] Add Tokens-to-Token ViT backbone and converted checkpoints. (#467) 2021-10-29 10:37:16 +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 [Feature] Add Twins backbone and convert checkpoints. (#642) 2022-01-27 10:25:05 +08:00
twins_svt_base.py [Feature] Add Twins backbone and convert checkpoints. (#642) 2022-01-27 10:25:05 +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