diff --git a/ppcls/arch/backbone/model_zoo/mobilevit_v2.py b/ppcls/arch/backbone/model_zoo/mobilevit_v2.py index d2a362f742611fda0d89be6b3626309b1b605bd4..ca8a550fc968bd9aa0801b29d506e8c9f147df00 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 8a17fdac6a3895a1bee3274b5595b2ddd4f1f5a2..ebbb8bf128796fbeb0b398c8a8ae9617a146cb19 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())