From d45e50b9db3745c1561dfae1813401576f60e336 Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Wed, 5 May 2021 17:51:23 -0700 Subject: [PATCH] Update test for cait 448x448 model --- tests/test_models.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tests/test_models.py b/tests/test_models.py index 96c51589..ced2fd76 100644 --- a/tests/test_models.py +++ b/tests/test_models.py @@ -43,7 +43,9 @@ def test_model_forward(model_name, batch_size): input_size = model.default_cfg['input_size'] if any([x > MAX_FWD_SIZE for x in input_size]): - # cap forward test at max res 448 * 448 to keep resource down + if is_model_default_key(model_name, 'fixed_input_size'): + pytest.skip("Fixed input size model > limit.") + # cap forward test at max res 384 * 384 to keep resource down input_size = tuple([min(x, MAX_FWD_SIZE) for x in input_size]) inputs = torch.randn((batch_size, *input_size)) outputs = model(inputs)