add dbot/crossmae refined models to torchhub

ml-jku · Jun 4, 2024 · 05441c0 · 05441c0
1 parent be49d6f
commit 05441c0
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -28,6 +28,9 @@ model = torch.hub.load("ml-jku/MIM-Refiner", "mae_refined_h14")
 model = torch.hub.load("ml-jku/MIM-Refiner", "mae_refined_twob14")
 model = torch.hub.load("ml-jku/MIM-Refiner", "d2v2_refined_l16")
 model = torch.hub.load("ml-jku/MIM-Refiner", "d2v2_refined_h14")
+model = torch.hub.load("ml-jku/MIM-Refiner", "dbot_refined_l16")
+model = torch.hub.load("ml-jku/MIM-Refiner", "dbot_refined_h14")
+model = torch.hub.load("ml-jku/MIM-Refiner", "crossmae_refined_l16")
 ```
 
 An example how to use torchhub models for a k-NN classifier can be

diff --git a/hubconf.py b/hubconf.py
@@ -25,6 +25,16 @@
         ctor_kwargs=VIT_CONFIGS["l16"],
         url="https://ml.jku.at/research/mimrefiner/download/maerefined_l16.th",
     ),
+    "dbot_refined_l16": dict(
+        ctor=PrenormVit,
+        ctor_kwargs=VIT_CONFIGS["l16"],
+        url="https://ml.jku.at/research/mimrefiner/download/dbotrefined_l16.th",
+    ),
+    "crossmae_refined_l16": dict(
+        ctor=PrenormVit,
+        ctor_kwargs=VIT_CONFIGS["l16"],
+        url="https://ml.jku.at/research/mimrefiner/download/crossmaerefined_l16.th",
+    ),
     "d2v2_refined_l16": dict(
         ctor=PostnormVit,
         ctor_kwargs=VIT_CONFIGS["l16"],
@@ -35,6 +45,11 @@
         ctor_kwargs=VIT_CONFIGS["h14"],
         url="https://ml.jku.at/research/mimrefiner/download/maerefined_h14.th",
     ),
+    "dbot_refined_h14": dict(
+        ctor=PrenormVit,
+        ctor_kwargs=VIT_CONFIGS["h14"],
+        url="https://ml.jku.at/research/mimrefiner/download/dbotrefined_h14.th",
+    ),
     "d2v2_refined_h14": dict(
         ctor=PostnormVit,
         ctor_kwargs=VIT_CONFIGS["h14"],