diff --git a/tests/test_models.py b/tests/test_models.py index b29fc639..c4ba8db3 100644 --- a/tests/test_models.py +++ b/tests/test_models.py @@ -69,7 +69,6 @@ def test_model_backward(model_name, batch_size): @pytest.mark.timeout(120) @pytest.mark.parametrize('model_name', list_models()) -#@pytest.mark.parametrize('model_name', ["xception41"]) @pytest.mark.parametrize('batch_size', [1]) def test_model_default_cfgs(model_name, batch_size): """Run a single forward pass with each model"""