Skip to content

Commit b10310c

Browse files
committed
Add proper pool size for new resnexts
1 parent 3561f8e commit b10310c

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

timm/models/resnet.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1054,11 +1054,11 @@ def _gcfg(url='', **kwargs):
10541054
first_conv='conv1.0', test_crop_pct=1.0),
10551055
'seresnextaa201d_32x8d.sw_in12k_ft_in1k_384': _cfg(
10561056
hf_hub_id='timm/',
1057-
interpolation='bicubic', first_conv='conv1.0', input_size=(3, 384, 384), crop_pct=1.0),
1057+
interpolation='bicubic', first_conv='conv1.0', pool_size=(12, 12), input_size=(3, 384, 384), crop_pct=1.0),
10581058
'seresnextaa201d_32x8d.sw_in12k': _cfg(
10591059
hf_hub_id='timm/',
10601060
num_classes=11821, interpolation='bicubic', first_conv='conv1.0',
1061-
crop_pct=0.95, input_size=(3, 320, 320), test_input_size=(3, 384, 384), test_crop_pct=1.0),
1061+
crop_pct=0.95, input_size=(3, 320, 320), pool_size=(10, 10), test_input_size=(3, 384, 384), test_crop_pct=1.0),
10621062

10631063
'resnetaa50d.sw_in12k': _ttcfg(
10641064
hf_hub_id='timm/',

0 commit comments

Comments
 (0)