Delete unnecessary files

a5a3b0f5 · weixin_46524038 · cuicheng01 · 272e7536 · 272e7536 · 272e7536
3 changed file
--- a/ppcls/arch/backbone/legendary_models/swin_transformerv2.py
+++ b/ppcls/arch/backbone/legendary_models/swin_transformerv2.py
--- a/ppcls/arch/backbone/model_zoo/swin_transformerv2.py
+++ b/ppcls/arch/backbone/model_zoo/swin_transformerv2.py
--- a/ppcls/utils/save_load.py
+++ b/ppcls/utils/save_load.py
@@ -64,10 +64,20 @@ def load_dygraph_pretrain(model, path=None):
    return


-def load_dygraph_pretrain_from_url(model, pretrained_url, use_ssld=False):
+def load_dygraph_pretrain_from_url(model,
+                                   pretrained_url,
+                                   use_ssld=False,
+                                   use_imagenet22k_pretrained=False,
+                                   use_imagenet22kto1k_pretrained=False):
    if use_ssld:
        pretrained_url = pretrained_url.replace("_pretrained",
                                                "_ssld_pretrained")
+    if use_imagenet22k_pretrained:
+        pretrained_url = pretrained_url.replace("_pretrained",
+                                                "_22k_pretrained")
+    if use_imagenet22kto1k_pretrained:
+        pretrained_url = pretrained_url.replace("_pretrained",
+                                                "_22kto1k_pretrained")
    local_weight_path = get_weights_path_from_url(pretrained_url).replace(
        ".pdparams", "")
    load_dygraph_pretrain(model, path=local_weight_path)