renamed ssl models from torch_hub and updated patch version

ViCCo-Group · Nov 17, 2022 · 239ba2e · 239ba2e
1 parent 63eb85d
commit 239ba2e
Show file tree

Hide file tree

Showing 5 changed files with 4 additions and 4 deletions.
diff --git a/thingsvision/_version.py b/thingsvision/_version.py
@@ -1 +1 @@
-__version__ = "2.2.7"
+__version__ = "2.2.10"
diff --git a/thingsvision/custom_models/__init__.py b/thingsvision/custom_models/__init__.py
@@ -1,8 +1,8 @@
 from .alexnet_ecoset import Alexnet_ecoset
-from .barlowtwins import BarlowTwins
+from .barlowtwins_rn50 import BarlowTwins
 from .resnet50_ecoset import Resnet50_ecoset
-from .swav import Swav
-from .vicreg import Vicreg
+from .swav_rn50 import Swav
+from .vicreg_rn50 import Vicreg
 from .vgg16_ecoset import VGG16_ecoset
 from .openclip import OpenCLIP
 from .inception_ecoset import Inception_ecoset

diff --git a/thingsvision/custom_models/barlowtwins.py → ...svision/custom_models/barlowtwins_rn50.py b/thingsvision/custom_models/barlowtwins.py → ...svision/custom_models/barlowtwins_rn50.py
diff --git a/thingsvision/custom_models/swav.py → thingsvision/custom_models/swav_rn50.py b/thingsvision/custom_models/swav.py → thingsvision/custom_models/swav_rn50.py
diff --git a/thingsvision/custom_models/vicreg.py → thingsvision/custom_models/vicreg_rn50.py b/thingsvision/custom_models/vicreg.py → thingsvision/custom_models/vicreg_rn50.py