Fix the bug in pipeline for dygraph mode (PaddlePaddle#32716)

* update, test=develop
sandyhouse · May 3, 2021 · 7438c81 · 7438c81
1 parent 6a1957e
commit 7438c81
Show file tree

Hide file tree

Showing 3 changed files with 231 additions and 155 deletions.
diff --git a/python/paddle/distributed/fleet/meta_parallel/parallel_layers/pp_layers.py b/python/paddle/distributed/fleet/meta_parallel/parallel_layers/pp_layers.py
@@ -108,7 +108,6 @@ def __init__(self,
         # construct layer
         self.run_function = []
         self._build_layer()
-        self.to(paddle.CUDAPlace(self.device_id))
 
     def _segment_network(self, seg_method):
         logger.info("start segment network..")