add to __init__ and api doc

open-mmlab · Aug 17, 2023 · 5ba8782 · 5ba8782
1 parent 857cf7a
commit 5ba8782
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 66 deletions.
diff --git a/docs/en/api/models.rst b/docs/en/api/models.rst
@@ -220,6 +220,8 @@ Backbones
    ViTSAM
    XCiT
    ViTEVA02
+   PromptedViT
+   PromptedViTEVA02
 
 .. module:: mmpretrain.models.necks
 

diff --git a/mmpretrain/models/backbones/__init__.py b/mmpretrain/models/backbones/__init__.py
@@ -57,73 +57,21 @@
 from .vision_transformer import VisionTransformer
 from .vit_eva02 import ViTEVA02
 from .vit_sam import ViTSAM
+from .vpt import PromptedViT, PromptedViTEVA02
 from .xcit import XCiT
 
 __all__ = [
-    'LeNet5',
-    'AlexNet',
-    'VGG',
-    'RegNet',
-    'ResNet',
-    'ResNeXt',
-    'ResNetV1d',
-    'ResNeSt',
-    'ResNet_CIFAR',
-    'SEResNet',
-    'SEResNeXt',
-    'ShuffleNetV1',
-    'ShuffleNetV2',
-    'MobileNetV2',
-    'MobileNetV3',
-    'VisionTransformer',
-    'SwinTransformer',
-    'TNT',
-    'TIMMBackbone',
-    'T2T_ViT',
-    'Res2Net',
-    'RepVGG',
-    'Conformer',
-    'MlpMixer',
-    'DistilledVisionTransformer',
-    'PCPVT',
-    'SVT',
-    'EfficientNet',
-    'EfficientNetV2',
-    'ConvNeXt',
-    'HRNet',
-    'ResNetV1c',
-    'ConvMixer',
-    'EdgeNeXt',
-    'CSPDarkNet',
-    'CSPResNet',
-    'CSPResNeXt',
-    'CSPNet',
-    'RepLKNet',
-    'RepMLPNet',
-    'PoolFormer',
-    'RIFormer',
-    'DenseNet',
-    'VAN',
-    'InceptionV3',
-    'MobileOne',
-    'EfficientFormer',
-    'SwinTransformerV2',
-    'MViT',
-    'DeiT3',
-    'HorNet',
-    'MobileViT',
-    'DaViT',
-    'BEiTViT',
-    'RevVisionTransformer',
-    'MixMIMTransformer',
-    'TinyViT',
-    'LeViT',
-    'Vig',
-    'PyramidVig',
-    'XCiT',
-    'ViTSAM',
-    'ViTEVA02',
-    'HiViT',
-    'SparseResNet',
-    'SparseConvNeXt',
+    'LeNet5', 'AlexNet', 'VGG', 'RegNet', 'ResNet', 'ResNeXt', 'ResNetV1d',
+    'ResNeSt', 'ResNet_CIFAR', 'SEResNet', 'SEResNeXt', 'ShuffleNetV1',
+    'ShuffleNetV2', 'MobileNetV2', 'MobileNetV3', 'VisionTransformer',
+    'SwinTransformer', 'TNT', 'TIMMBackbone', 'T2T_ViT', 'Res2Net', 'RepVGG',
+    'Conformer', 'MlpMixer', 'DistilledVisionTransformer', 'PCPVT', 'SVT',
+    'EfficientNet', 'EfficientNetV2', 'ConvNeXt', 'HRNet', 'ResNetV1c',
+    'ConvMixer', 'EdgeNeXt', 'CSPDarkNet', 'CSPResNet', 'CSPResNeXt', 'CSPNet',
+    'RepLKNet', 'RepMLPNet', 'PoolFormer', 'RIFormer', 'DenseNet', 'VAN',
+    'InceptionV3', 'MobileOne', 'EfficientFormer', 'SwinTransformerV2', 'MViT',
+    'DeiT3', 'HorNet', 'MobileViT', 'DaViT', 'BEiTViT', 'RevVisionTransformer',
+    'MixMIMTransformer', 'TinyViT', 'LeViT', 'Vig', 'PyramidVig', 'XCiT',
+    'ViTSAM', 'ViTEVA02', 'HiViT', 'SparseResNet', 'SparseConvNeXt',
+    'PromptedViT', 'PromptedViTEVA02'
 ]