Update sam_presets.py (keras-team#1940)

SamanehSaadat · Oct 18, 2024 · 2502d3f · 2502d3f
1 parent efedb6f
commit 2502d3f
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/keras_hub/src/models/sam/sam_presets.py b/keras_hub/src/models/sam/sam_presets.py
@@ -9,7 +9,7 @@
             "path": "sam",
             "model_card": "https://arxiv.org/abs/2304.02643",
         },
-        "kaggle_handle": "kaggle://keras/sam/keras/sam_base_sa1b/2",
+        "kaggle_handle": "kaggle://keras/sam/keras/sam_base_sa1b/4",
     },
     "sam_large_sa1b": {
         "metadata": {
@@ -19,7 +19,7 @@
             "path": "sam",
             "model_card": "https://arxiv.org/abs/2304.02643",
         },
-        "kaggle_handle": "kaggle://keras/sam/keras/sam_large_sa1b/3",
+        "kaggle_handle": "kaggle://keras/sam/keras/sam_large_sa1b/4",
     },
     "sam_huge_sa1b": {
         "metadata": {
@@ -29,6 +29,6 @@
             "path": "sam",
             "model_card": "https://arxiv.org/abs/2304.02643",
         },
-        "kaggle_handle": "kaggle://keras/sam/keras/sam_huge_sa1b/3",
+        "kaggle_handle": "kaggle://keras/sam/keras/sam_huge_sa1b/4",
     },
 }