From 08d60f4a9a385b1eaf132a0ac1cff37b4ad94ac8 Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Tue, 4 May 2021 12:41:09 -0700 Subject: [PATCH] resnetrs50 pool sizing wrong --- 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 5355d61d..cccc9ae0 100644 --- a/timm/models/resnet.py +++ b/timm/models/resnet.py @@ -242,7 +242,7 @@ default_cfgs = { # ResNet-RS models 'resnetrs50': _cfg( url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-rs-weights/resnetrs50-7c9728e2.pth', - input_size=(3, 160, 160), pool_size=(4, 4), crop_pct=0.91, test_input_size=(3, 224, 224), + input_size=(3, 160, 160), pool_size=(5, 5), crop_pct=0.91, test_input_size=(3, 224, 224), interpolation='bicubic', first_conv='conv1.0'), 'resnetrs101': _cfg( url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-rs-weights/resnetrs101-3e4bb55c.pth',