Skip to content
Snippets Groups Projects
Commit e3e41ba6 authored by Neta Zmora's avatar Neta Zmora
Browse files

Extend pruning tests to parallel models

parent b64be690
No related branches found
No related tags found
No related merge requests found
...@@ -83,7 +83,7 @@ def vgg19_imagenet(is_parallel): ...@@ -83,7 +83,7 @@ def vgg19_imagenet(is_parallel):
("features.32", "features.34")], ("features.32", "features.34")],
bn_name=None) bn_name=None)
@pytest.fixture(params=[False]) @pytest.fixture(params=[True, False])
def parallel(request): def parallel(request):
return request.param return request.param
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment