Skip to content

Commit a81f5e0

Browse files
committed
cherrypick for #32640 :add profile and fix dataset hang in heterps;test=develop
1 parent 6e02a8d commit a81f5e0

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

python/paddle/distributed/fleet/dataset/dataset.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -541,7 +541,7 @@ def _dynamic_adjust_before_train(self, thread_num):
541541
if self.use_ps_gpu:
542542
self.dataset.dynamic_adjust_channel_num(thread_num, True)
543543
else:
544-
self.dataset.dynamic_adjust_channel_num(thread_num, False)
544+
self.dataset.dynamic_adjust_channel_num(thread_num, False)
545545
self.dataset.dynamic_adjust_readers_num(thread_num)
546546

547547
def _dynamic_adjust_after_train(self):

python/paddle/fluid/dataset.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -403,7 +403,7 @@ def _dynamic_adjust_before_train(self, thread_num):
403403
if self.use_ps_gpu:
404404
self.dataset.dynamic_adjust_channel_num(thread_num, True)
405405
else:
406-
self.dataset.dynamic_adjust_channel_num(thread_num, False)
406+
self.dataset.dynamic_adjust_channel_num(thread_num, False)
407407
self.dataset.dynamic_adjust_readers_num(thread_num)
408408

409409
@deprecated(
@@ -415,7 +415,7 @@ def _dynamic_adjust_after_train(self):
415415
if self.use_ps_gpu:
416416
self.dataset.dynamic_adjust_channel_num(self.thread_num, True)
417417
else:
418-
self.dataset.dynamic_adjust_channel_num(self.thread_num, False)
418+
self.dataset.dynamic_adjust_channel_num(self.thread_num, False)
419419
self.dataset.dynamic_adjust_readers_num(self.thread_num)
420420

421421
@deprecated(

0 commit comments

Comments
 (0)