Add min_input_size to bat_resnext to prevent test breakage.

3 years ago · b3b90d944d
parent 758c4438a7
commit b3b90d944d
1 changed files with 2 additions and 1 deletions
--- a/timm/models/byobnet.py
+++ b/timm/models/byobnet.py
@ -102,7 +102,8 @@ default_cfgs = {
    'gcresnext26ts': _cfg(
        first_conv='stem.conv1.conv', input_size=(3, 256, 256), pool_size=(8, 8), interpolation='bicubic'),
    'bat_resnext26ts': _cfg(
-        first_conv='stem.conv1.conv', input_size=(3, 256, 256), pool_size=(8, 8), interpolation='bicubic'),
+        first_conv='stem.conv1.conv', input_size=(3, 256, 256), pool_size=(8, 8), interpolation='bicubic',
+        min_input_size=(3, 128, 128)),
 }