v0.0.2

ffiirree · Nov 28, 2021 · 5ea3663 · 5ea3663
1 parent 518aa17
commit 5ea3663
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/cvm/utils/utils.py b/cvm/utils/utils.py
@@ -184,8 +184,8 @@ def list_models(lib: str = 'all'):
     assert lib in ['all', 'cvm', 'torch', 'timm'], f'Unknown library {lib}.'
 
     if lib == 'all':
-        cvm_models = _filter_models(torchvision.models.__dict__, sort=True)
-        torch_models = _filter_models(models.__dict__, 'torch/', True)
+        cvm_models = _filter_models(models.__dict__, sort=True)
+        torch_models = _filter_models(torchvision.models.__dict__, 'torch/', sort=True)
 
         timm_models = [
             'timm/' + name for name in timm.list_models()

diff --git a/cvm/version.py b/cvm/version.py
@@ -1 +1 @@
-__version__ = '0.0.1'
+__version__ = '0.0.2'