Skip to content

Commit 0020268

Browse files
committed
Try lower max size for non_std default_cfg test
1 parent 4d96165 commit 0020268

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/test_models.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -174,7 +174,7 @@ def test_model_default_cfgs_non_std(model_name, batch_size):
174174
cfg = model.default_cfg
175175

176176
input_size = _get_input_size(model_name=model_name, target=TARGET_FWD_SIZE)
177-
if max(input_size) > MAX_FWD_SIZE:
177+
if max(input_size) > 320: # FIXME const
178178
pytest.skip("Fixed input size model > limit.")
179179

180180
input_tensor = torch.randn((batch_size, *input_size))

0 commit comments

Comments
 (0)