minor fix in IT

Digital-Dermatology · Mar 19, 2024 · ad19822 · ad19822
1 parent f19e50f
commit ad19822
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/cleaner/selfclean.py b/src/cleaner/selfclean.py
@@ -63,7 +63,7 @@
 
 class PretrainingType(Enum):
     IMAGENET = "imagenet"
-    IMAGENET_VIT = "imagenet_vit"
+    IMAGENET_VIT = "imagenet_vit_tiny"
     DINO = "dino"
 
 

diff --git a/ssl_library b/ssl_library