fix parallel strategy redefine (#258)

42a8767b · Bai Yifan · GitHub · a07b5c20 · 42a8767b
隐藏空白更改
内联并排

Showing with 0 addition and 2 deletion

paddleslim/nas/darts/train_search.py paddleslim/nas/darts/train_search.py +0 -2

未找到文件。
--- a/paddleslim/nas/darts/train_search.py
+++ b/paddleslim/nas/darts/train_search.py
@@ -176,8 +176,6 @@ class DARTSearch(object):

        """

-        if self.use_data_parallel:
-            strategy = fluid.dygraph.parallel.prepare_context()
        model_parameters = [
            p for p in self.model.parameters()
            if p.name not in [a.name for a in self.model.arch_parameters()]