|
|
@ -69,7 +69,6 @@ def test_model_backward(model_name, batch_size):
|
|
|
|
|
|
|
|
|
|
|
|
@pytest.mark.timeout(120)
|
|
|
|
@pytest.mark.timeout(120)
|
|
|
|
@pytest.mark.parametrize('model_name', list_models())
|
|
|
|
@pytest.mark.parametrize('model_name', list_models())
|
|
|
|
#@pytest.mark.parametrize('model_name', ["xception41"])
|
|
|
|
|
|
|
|
@pytest.mark.parametrize('batch_size', [1])
|
|
|
|
@pytest.mark.parametrize('batch_size', [1])
|
|
|
|
def test_model_default_cfgs(model_name, batch_size):
|
|
|
|
def test_model_default_cfgs(model_name, batch_size):
|
|
|
|
"""Run a single forward pass with each model"""
|
|
|
|
"""Run a single forward pass with each model"""
|
|
|
|