Skip to content

Commit b3b90d9

Browse files
committed
Add min_input_size to bat_resnext to prevent test breakage.
1 parent 758c443 commit b3b90d9

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

timm/models/byobnet.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -102,7 +102,8 @@ def _cfg(url='', **kwargs):
102102
'gcresnext26ts': _cfg(
103103
first_conv='stem.conv1.conv', input_size=(3, 256, 256), pool_size=(8, 8), interpolation='bicubic'),
104104
'bat_resnext26ts': _cfg(
105-
first_conv='stem.conv1.conv', input_size=(3, 256, 256), pool_size=(8, 8), interpolation='bicubic'),
105+
first_conv='stem.conv1.conv', input_size=(3, 256, 256), pool_size=(8, 8), interpolation='bicubic',
106+
min_input_size=(3, 128, 128)),
106107
}
107108

108109

0 commit comments

Comments
 (0)