Update test_inference.py

Not six min
2020-05-07 10:15:49 -07:00 · 2020-05-07 10:15:49 -07:00 · e545bb9401
parent 305a2db705
commit e545bb9401
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(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):