cherrypick for PaddlePaddle#32640 :add profile and fix dataset hang i…

…n heterps;test=develop
danleifeng · May 18, 2021 · 6e02a8d · 6e02a8d
1 parent 7166cd0
commit 6e02a8d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/python/paddle/distributed/fleet/dataset/dataset.py b/python/paddle/distributed/fleet/dataset/dataset.py
@@ -221,7 +221,7 @@ def _set_use_ps_gpu(self, use_ps_gpu):
         """
         self.use_ps_gpu = use_ps_gpu
 
-     def _finish_to_run(self):
+    def _finish_to_run(self):
         self.dataset.destroy_readers()
 
     def _desc(self):

diff --git a/python/paddle/fluid/executor.py b/python/paddle/fluid/executor.py
@@ -1510,7 +1510,7 @@ def _run_from_dataset(self,
         # in case of calling _set_use_ps_gpu explicitly
         if dataset.use_ps_gpu is False:
             dataset._set_use_ps_gpu(trainer.proto_desc.use_ps_gpu)
-	dataset._dynamic_adjust_before_train(trainer.proto_desc.thread_num)
+        dataset._dynamic_adjust_before_train(trainer.proto_desc.thread_num)
 
         trainer_instance = self._default_executor.init_for_dataset(
             program.desc, trainer._desc(), scope, dataset.dataset)