提交 e8bf35d1 编写于 作者: Z zh-hike 提交者: zengshao0622

fix foundationvit load fail

上级 83fc10cd
......@@ -71,6 +71,7 @@ from .model_zoo.mobilevit import MobileViT_XXS, MobileViT_XS, MobileViT_S
from .model_zoo.repvgg import RepVGG_A0, RepVGG_A1, RepVGG_A2, RepVGG_B0, RepVGG_B1, RepVGG_B2, RepVGG_B1g2, RepVGG_B1g4, RepVGG_B2g4, RepVGG_B3, RepVGG_B3g4, RepVGG_D2se
from .model_zoo.van import VAN_B0, VAN_B1, VAN_B2, VAN_B3
from .model_zoo.peleenet import PeleeNet
from .model_zoo.foundation_vit import CLIP_base_patch32_224, CLIP_base_patch16_224, CLIP_large_patch14_336, CLIP_large_patch14_224, BEiTv2_base_patch16_224, BEiTv2_large_patch16_224, CAE_base_patch16_224, EVA_huge_patch14, MOCOV3_small, MOCOV3_base, MAE_huge_patch14, MAE_large_patch16, MAE_base_patch16
from .model_zoo.convnext import ConvNeXt_tiny, ConvNeXt_small, ConvNeXt_base_224, ConvNeXt_base_384, ConvNeXt_large_224, ConvNeXt_large_384
from .model_zoo.nextvit import NextViT_small_224, NextViT_base_224, NextViT_large_224, NextViT_small_384, NextViT_base_384, NextViT_large_384
from .model_zoo.cae import cae_base_patch16_224, cae_large_patch16_224
......
......@@ -26,7 +26,6 @@ from paddle.nn.initializer import TruncatedNormal, Constant, Normal
from ....utils.save_load import load_dygraph_pretrain, load_dygraph_pretrain_from_url
MODEL_URLS = {
"CLIP_small_patch16_224": None,
"CLIP_base_patch32_224": None,
"CLIP_base_patch16_224": None,
"CLIP_large_patch14_336": None,
......@@ -34,7 +33,7 @@ MODEL_URLS = {
"BEiTv2_base_patch16_224": None,
"BEiTv2_large_patch16_224": None,
"CAE_base_patch16_224": None,
"EVA_small_patch16_224": None,
'EVA_huge_patch14':None,
"MOCOV3_small": None,
"MOCOV3_base": None,
"MAE_huge_patch14": None,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册