diff --git a/tests/test_pruning.py b/tests/test_pruning.py index 5007cd3d7b6a26fc2f4906e0af80787799238faa..7b26709ec2b549392e5ee5f00d2299215420efcf 100755 --- a/tests/test_pruning.py +++ b/tests/test_pruning.py @@ -83,7 +83,7 @@ def vgg19_imagenet(is_parallel): ("features.32", "features.34")], bn_name=None) -@pytest.fixture(params=[False]) +@pytest.fixture(params=[True, False]) def parallel(request): return request.param