v0.0.3

ffiirree · Nov 30, 2021 · ecb614b · ecb614b
1 parent 9552edc
commit ecb614b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/cvm/models/core/utils.py b/cvm/models/core/utils.py
@@ -36,4 +36,5 @@ def load_from_local_or_url(model, pth=None, url=None, progress=True):
         state_dict = torch.load(os.path.expanduser(pth))
     else:
         state_dict = torch.hub.load_state_dict_from_url(url, progress=progress)
-        model.load_state_dict(state_dict)
+
+    model.load_state_dict(state_dict)
diff --git a/cvm/version.py b/cvm/version.py
@@ -1 +1 @@
-__version__ = '0.0.2'
+__version__ = '0.0.3'