diff --git a/tests/test_inference.py b/tests/test_inference.py index 34bac63d..dc45c409 100644 --- a/tests/test_inference.py +++ b/tests/test_inference.py @@ -4,7 +4,7 @@ import torch from timm import list_models, create_model -@pytest.mark.timeout(360) +@pytest.mark.timeout(300) @pytest.mark.parametrize('model_name', list_models()) @pytest.mark.parametrize('batch_size', [1]) def test_model_forward(model_name, batch_size):