From 305a2db70566bb991a1465c41f4c298a8bcb8a77 Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Thu, 7 May 2020 10:14:24 -0700 Subject: [PATCH] Update test_inference.py Make the timeout 5-min for now, see if we can get a pass... --- tests/test_inference.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_inference.py b/tests/test_inference.py index 75b8d445..34bac63d 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(60) +@pytest.mark.timeout(360) @pytest.mark.parametrize('model_name', list_models()) @pytest.mark.parametrize('batch_size', [1]) def test_model_forward(model_name, batch_size):