From 0020268d9b292a3b8ac82dcb1e21039ca32b0823 Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Sat, 12 Jun 2021 23:31:24 -0700 Subject: [PATCH] Try lower max size for non_std default_cfg test --- tests/test_models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_models.py b/tests/test_models.py index ac156806..52a8023a 100644 --- a/tests/test_models.py +++ b/tests/test_models.py @@ -174,7 +174,7 @@ def test_model_default_cfgs_non_std(model_name, batch_size): cfg = model.default_cfg input_size = _get_input_size(model_name=model_name, target=TARGET_FWD_SIZE) - if max(input_size) > MAX_FWD_SIZE: + if max(input_size) > 320: # FIXME const pytest.skip("Fixed input size model > limit.") input_tensor = torch.randn((batch_size, *input_size))