Add overlapped stem convnext zepto weights

huggingface · Oct 8, 2024 · 41a79e0 · 41a79e0
1 parent 4d4bdd6
commit 41a79e0
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/timm/models/convnext.py b/timm/models/convnext.py
@@ -599,10 +599,10 @@ def _cfgv2(url='', **kwargs):
     '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/',
+    'convnext_zepto_rms_ols.ra4_e3600_r224_in1k': _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),
+        crop_pct=0.9),
     'convnext_atto.d2_in1k': _cfg(
         url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-rsb-weights/convnext_atto_d2-01bb0f51.pth',
         hf_hub_id='timm/',