From e3e41ba6205cd50a45d130870d4cbbff621130a8 Mon Sep 17 00:00:00 2001
From: Neta Zmora <neta.zmora@intel.com>
Date: Wed, 11 Jul 2018 17:25:43 +0300
Subject: [PATCH] Extend pruning tests to parallel models
---
tests/test_pruning.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/tests/test_pruning.py b/tests/test_pruning.py
index 5007cd3..7b26709 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
--
GitLab