From b3b90d944d7c77c34428225ecd25c4d55761c20e Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Tue, 8 Jun 2021 17:32:08 -0700 Subject: [PATCH] Add min_input_size to bat_resnext to prevent test breakage. --- timm/models/byobnet.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/timm/models/byobnet.py b/timm/models/byobnet.py index 750a43de..692cea8c 100644 --- 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)), }