Update test_inference.py

Make the timeout 5-min for now, see if we can get a pass...
5 years ago · 305a2db705
parent 14e01b878c
commit 305a2db705
1 changed files with 1 additions and 1 deletions
--- 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):