mirrors
/
PaddleClas
mirror of https://github.com/PaddlePaddle/PaddleClas.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

update pretrained url

pull/2832/head
gaotingquan 2023-05-06 15:43:00 +00:00 committed by cuicheng01
parent e1a4c80ad3
commit 4643fdee09
2 changed files with 24 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
ppcls/arch/backbone/model_zoo/mobilevit_v2.py
View File

@ -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)

15
ppcls/arch/backbone/model_zoo/tinynet.py
View File

@ -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())