From db46dbd8fea2c5b839f4be8d560a5c2eef2873c0 Mon Sep 17 00:00:00 2001 From: cuicheng01 Date: Mon, 12 Jul 2021 10:35:53 +0000 Subject: [PATCH] Update save_load.py --- ppcls/utils/save_load.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ppcls/utils/save_load.py b/ppcls/utils/save_load.py index cca79f4c..625a2848 100644 --- a/ppcls/utils/save_load.py +++ b/ppcls/utils/save_load.py @@ -54,7 +54,7 @@ def load_dygraph_pretrain(model, path=None): return -def load_dygraph_pretrain_from_url(model, pretrained_url, use_ssld): +def load_dygraph_pretrain_from_url(model, pretrained_url, use_ssld=False): if use_ssld: pretrained_url = pretrained_url.replace("_pretrained", "_ssld_pretrained") -- GitLab