From 66be3a0cbeb8b2d29be7eddd7cbbdd06a21cf757 Mon Sep 17 00:00:00 2001 From: Bai Yifan Date: Thu, 7 May 2020 20:47:18 +0800 Subject: [PATCH] fix parallel strategy redefine (#258) (#259) --- paddleslim/nas/darts/train_search.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/paddleslim/nas/darts/train_search.py b/paddleslim/nas/darts/train_search.py index 74d1b7ee..056944e6 100644 --- 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()] -- GitLab