Mercurial > repos > goeckslab > image_learner

diff constants.py @ 3:2c3a3dfaf1a9 draft default tip

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
planemo upload for repository https://github.com/goeckslab/gleam.git commit 5ab02b6688c9af85525dd225134db8f2bbed76ed
author goeckslab
date Fri, 04 Jul 2025 03:44:56 +0000
parents 186424a7eca7
children
line wrap: on
line diff
--- a/constants.py	Thu Jul 03 20:43:24 2025 +0000
+++ b/constants.py	Fri Jul 04 03:44:56 2025 +0000
@@ -68,11 +68,11 @@
     "swin_v2_t": {"type": "swin_transformer", "model_variant": "v2_t"},
     "swin_v2_s": {"type": "swin_transformer", "model_variant": "v2_s"},
     "swin_v2_b": {"type": "swin_transformer", "model_variant": "v2_b"},
-    "vit_b_16": {"type": "vision_transformer", "model_variant": "b_16"},
-    "vit_b_32": {"type": "vision_transformer", "model_variant": "b_32"},
-    "vit_l_16": {"type": "vision_transformer", "model_variant": "l_16"},
-    "vit_l_32": {"type": "vision_transformer", "model_variant": "l_32"},
-    "vit_h_14": {"type": "vision_transformer", "model_variant": "h_14"},
+    "vit_b_16": {"type": "vit", "model_variant": "b_16"},
+    "vit_b_32": {"type": "vit", "model_variant": "b_32"},
+    "vit_l_16": {"type": "vit", "model_variant": "l_16"},
+    "vit_l_32": {"type": "vit", "model_variant": "l_32"},
+    "vit_h_14": {"type": "vit", "model_variant": "h_14"},
     "convnext_tiny": {"type": "convnext", "model_variant": "tiny"},
     "convnext_small": {"type": "convnext", "model_variant": "small"},
     "convnext_base": {"type": "convnext", "model_variant": "base"},
@@ -80,10 +80,10 @@
     "maxvit_t": {"type": "maxvit", "model_variant": "t"},
     "alexnet": {"type": "alexnet"},
     "googlenet": {"type": "googlenet"},
-    "inception_v3": {"type": "inception_v3"},
+    "inception_v3": {"type": "inceptionv3"},
     "mobilenet_v2": {"type": "mobilenet_v2"},
-    "mobilenet_v3_large": {"type": "mobilenet_v3_large"},
-    "mobilenet_v3_small": {"type": "mobilenet_v3_small"},
+    "mobilenet_v3_large": {"type": "mobilenetv3", "model_variant": "large"},
+    "mobilenet_v3_small": {"type": "mobilenetv3", "model_variant": "small"},
 }
 METRIC_DISPLAY_NAMES = {
     "accuracy": "Accuracy",