missed first_conv for rnv2 18d

huggingface · Oct 8, 2024 · 72f0edb · 72f0edb
1 parent 3ed603a
commit 72f0edb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/timm/models/resnetv2.py b/timm/models/resnetv2.py
@@ -699,7 +699,7 @@ def _cfg(url='', **kwargs):
     'resnetv2_18.untrained': _cfg(
         interpolation='bicubic', crop_pct=0.95),
     'resnetv2_18d.untrained': _cfg(
-        interpolation='bicubic', crop_pct=0.95),
+        interpolation='bicubic', crop_pct=0.95, first_conv='stem.conv1'),
     'resnetv2_50.a1h_in1k': _cfg(
         hf_hub_id='timm/',
         interpolation='bicubic', crop_pct=0.95, test_input_size=(3, 288, 288), test_crop_pct=1.0),