Missed first_conv from resnet18d

huggingface · Nov 1, 2024 · d4dde48 · d4dde48
1 parent e6263bf
commit d4dde48
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/timm/models/resnet.py b/timm/models/resnet.py
@@ -713,7 +713,7 @@ def _gcfg(url='', **kwargs):
         first_conv='conv1.0'),
     'resnet18d.ra4_e3600_r224_in1k': _rcfg(
         hf_hub_id='timm/',
-        mean=(0.5, 0.5, 0.5), std=(0.5, 0.5, 0.5), crop_pct=0.9),
+        mean=(0.5, 0.5, 0.5), std=(0.5, 0.5, 0.5), crop_pct=0.9, first_conv='conv1.0'),
     'resnet34.a1_in1k': _rcfg(
         hf_hub_id='timm/',
         url='https://github.com/huggingface/pytorch-image-models/releases/download/v0.1-rsb-weights/resnet34_a1_0-46f8f793.pth'),