From d4dde48dd516d3583128d9ba6548a62cccd93fbc Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Thu, 31 Oct 2024 19:29:53 -0700 Subject: [PATCH] Missed first_conv from resnet18d --- timm/models/resnet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/timm/models/resnet.py b/timm/models/resnet.py index 280463e1..3f8d6b1a 100644 --- a/timm/models/resnet.py +++ b/timm/models/resnet.py @@ -713,7 +713,7 @@ default_cfgs = generate_default_cfgs({ 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'),