update

PaddlePaddle · Apr 23, 2021 · 67d17ca · 67d17ca
1 parent aea875a
commit 67d17ca
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/python/paddle/hapi/hub.py b/python/paddle/hapi/hub.py
@@ -78,9 +78,8 @@ def _make_dirs(dirname):
 def _get_cache_or_reload(repo, force_reload, verbose=True, source='github'):
     # Setup hub_dir to save downloaded files
     hub_dir = HUB_DIR
-    if not os.path.exists(hub_dir):
-        # os.makedirs(hub_dir)
-        _make_dirs(hub_dir)
+
+    _make_dirs(hub_dir)
 
     # Parse github/gitee repo information
     repo_owner, repo_name, branch = _parse_repo_info(repo)

diff --git a/python/paddle/tests/test_hapi_hub.py b/python/paddle/tests/test_hapi_hub.py
@@ -47,6 +47,13 @@ def testLoad(self, ):
             force_reload=False,
             pretrained=False)
 
+        model = hub.load(
+            self.github_repo.split(':')[0],
+            model='MM',
+            source='github',
+            force_reload=False,
+            pretrained=False)
+
         model = hub.load(
             self.github_repo,
             model='MM',