From 52ac8814024e373590e13c192dae16846cdf97ae Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Fri, 22 Apr 2022 20:55:52 -0700 Subject: [PATCH] Missed first_conv in latest seresnext 'D' default_cfgs --- timm/models/resnet.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/timm/models/resnet.py b/timm/models/resnet.py index d304cf42..a7f0c0f6 100644 --- a/timm/models/resnet.py +++ b/timm/models/resnet.py @@ -243,7 +243,7 @@ default_cfgs = { interpolation='bicubic', test_input_size=(3, 288, 288), crop_pct=1.0), 'seresnext101d_32x8d': _cfg( url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-tpu-weights/seresnext101d_32x8d_ah-191d7b94.pth', - interpolation='bicubic', test_input_size=(3, 288, 288), crop_pct=1.0), + interpolation='bicubic', first_conv='conv1.0', test_input_size=(3, 288, 288), crop_pct=1.0), 'senet154': _cfg( url='', @@ -273,7 +273,7 @@ default_cfgs = { interpolation='bicubic', first_conv='conv1.0'), 'seresnextaa101d_32x8d': _cfg( url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-tpu-weights/seresnextaa101d_32x8d_ah-83c8ae12.pth', - interpolation='bicubic', test_input_size=(3, 288, 288), crop_pct=1.0), + interpolation='bicubic', first_conv='conv1.0', test_input_size=(3, 288, 288), crop_pct=1.0), # ResNet-RS models 'resnetrs50': _cfg(