Add zepto weights

huggingface · Sep 30, 2024 · 69b687d · 69b687d
1 parent c6e5557
commit 69b687d
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/timm/models/convnext.py b/timm/models/convnext.py
@@ -596,10 +596,9 @@ def _cfgv2(url='', **kwargs):
         hf_hub_id='timm/',
         crop_pct=0.95, test_input_size=(3, 288, 288), test_crop_pct=1.0),
 
-    'convnext_zepto_rms.untrained': _cfg(
-        #hf_hub_id='timm/',
-        mean=(0.5, 0.5, 0.5), std=(0.5, 0.5, 0.5),
-        test_input_size=(3, 256, 256), test_crop_pct=0.95),
+    'convnext_zepto_rms.ra4_e3600_r224_in1k': _cfg(
+        hf_hub_id='timm/',
+        mean=(0.5, 0.5, 0.5), std=(0.5, 0.5, 0.5)),
     'convnext_zepto_rms_ols.untrained': _cfg(
         # hf_hub_id='timm/',
         mean=(0.5, 0.5, 0.5), std=(0.5, 0.5, 0.5),