From 4643fdee0903f74132c5ad92560c5269b7e9341c Mon Sep 17 00:00:00 2001
From: gaotingquan <gaotingquan@baidu.com>
Date: Sat, 6 May 2023 15:43:00 +0000
Subject: [PATCH] update pretrained url

---
 ppcls/arch/backbone/model_zoo/mobilevit_v2.py | 21 ++++++++++++-------
 ppcls/arch/backbone/model_zoo/tinynet.py      | 15 ++++++++-----
 2 files changed, 24 insertions(+), 12 deletions(-)

diff --git a/ppcls/arch/backbone/model_zoo/mobilevit_v2.py b/ppcls/arch/backbone/model_zoo/mobilevit_v2.py
index d2a362f7..ca8a550f 100644
--- a/ppcls/arch/backbone/model_zoo/mobilevit_v2.py
+++ b/ppcls/arch/backbone/model_zoo/mobilevit_v2.py
@@ -25,13 +25,20 @@ import paddle.nn.functional as F
 from ....utils.save_load import load_dygraph_pretrain, load_dygraph_pretrain_from_url
 
 MODEL_URLS = {
-    "MobileViTV2_x0_5": "",
-    "MobileViTV2_x0_75": "",
-    "MobileViTV2_x1_0": "",
-    "MobileViTV2_x1_25": "",
-    "MobileViTV2_x1_5": "",
-    "MobileViTV2_x1_75": "",
-    "MobileViTV2_x2_0": "",
+    "MobileViTV2_x0_5":
+    "https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/MobileViTV2_x0_5_pretrained.pdparams",
+    "MobileViTV2_x0_75":
+    "https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/MobileViTV2_x0_75_pretrained.pdparams",
+    "MobileViTV2_x1_0":
+    "https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/MobileViTV2_x1_0_pretrained.pdparams",
+    "MobileViTV2_x1_25":
+    "https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/MobileViTV2_x1_25_pretrained.pdparams",
+    "MobileViTV2_x1_5":
+    "https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/MobileViTV2_x1_5_pretrained.pdparams",
+    "MobileViTV2_x1_75":
+    "https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/MobileViTV2_x1_75_pretrained.pdparams",
+    "MobileViTV2_x2_0":
+    "https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/MobileViTV2_x2_0_pretrained.pdparams",
 }
 
 layer_norm_2d = partial(nn.GroupNorm, num_groups=1)
diff --git a/ppcls/arch/backbone/model_zoo/tinynet.py b/ppcls/arch/backbone/model_zoo/tinynet.py
index 8a17fdac..ebbb8bf1 100644
--- a/ppcls/arch/backbone/model_zoo/tinynet.py
+++ b/ppcls/arch/backbone/model_zoo/tinynet.py
@@ -21,11 +21,16 @@ from .efficientnet import EfficientNet, efficientnet
 from ....utils.save_load import load_dygraph_pretrain, load_dygraph_pretrain_from_url
 
 MODEL_URLS = {
-    "TinyNet_A": "",
-    "TinyNet_B": "",
-    "TinyNet_C": "",
-    "TinyNet_D": "",
-    "TinyNet_E": "",
+    "TinyNet_A":
+    "https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/TinyNet_A_pretrained.pdparams",
+    "TinyNet_B":
+    "https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/TinyNet_B_pretrained.pdparams",
+    "TinyNet_C":
+    "https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/TinyNet_C_pretrained.pdparams",
+    "TinyNet_D":
+    "https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/TinyNet_D_pretrained.pdparams",
+    "TinyNet_E":
+    "https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/TinyNet_E_pretrained.pdparams",
 }
 
 __all__ = list(MODEL_URLS.keys())
-- 
GitLab