提交 4643fdee 编写于 作者: G gaotingquan 提交者: cuicheng01

update pretrained url

上级 e1a4c80a
......@@ -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)
......
......@@ -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())
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册