From 84049d7f1e19a7cbe03bd0f727a13b46da504576 Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Sun, 17 Nov 2024 12:44:08 -0800 Subject: [PATCH] Missed input_size pretraind_cfg metadata for v2 34d @ 384 --- timm/models/resnetv2.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/timm/models/resnetv2.py b/timm/models/resnetv2.py index fec91a49..18c1cdbb 100644 --- a/timm/models/resnetv2.py +++ b/timm/models/resnetv2.py @@ -712,8 +712,8 @@ default_cfgs = generate_default_cfgs({ first_conv='stem.conv1'), 'resnetv2_34d.ra4_e3600_r384_in1k': _cfg( hf_hub_id='timm/', - interpolation='bicubic', crop_pct=1.0, test_input_size=(3, 448, 448), pool_size=(12, 12), - first_conv='stem.conv1'), + crop_pct=1.0, input_size=(3, 384, 348), pool_size=(12, 12), test_input_size=(3, 448, 448), + interpolation='bicubic', 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),