From e8bf35d1b4653bd7869d676a4cc9f874897a0d5e Mon Sep 17 00:00:00 2001
From: zh-hike <1583124882@qq.com>
Date: Mon, 6 Feb 2023 12:08:28 +0000
Subject: [PATCH] fix foundationvit load fail

---
 ppcls/arch/backbone/__init__.py                 | 1 +
 ppcls/arch/backbone/model_zoo/foundation_vit.py | 3 +--
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/ppcls/arch/backbone/__init__.py b/ppcls/arch/backbone/__init__.py
index 2ffb38fc..c1f19313 100644
--- a/ppcls/arch/backbone/__init__.py
+++ b/ppcls/arch/backbone/__init__.py
@@ -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
diff --git a/ppcls/arch/backbone/model_zoo/foundation_vit.py b/ppcls/arch/backbone/model_zoo/foundation_vit.py
index 2814f349..1233db81 100644
--- a/ppcls/arch/backbone/model_zoo/foundation_vit.py
+++ b/ppcls/arch/backbone/model_zoo/foundation_vit.py
@@ -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,
-- 
GitLab