Skip to content

Commit a33a4f3

Browse files
authored
fix read-data timer when ignore_data_skip=False and skip_profile_timer=False (#8177)
1 parent d577e19 commit a33a4f3

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

paddlenlp/trainer/trainer.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -921,6 +921,7 @@ def _inner_training_loop(
921921
steps_trained_progress_bar.update(1)
922922
if steps_trained_in_current_epoch == 0:
923923
self._load_rng_state(resume_from_checkpoint)
924+
self.timers and self.timers("read-data").start()
924925
continue
925926
elif steps_trained_progress_bar is not None:
926927
steps_trained_progress_bar.close()

0 commit comments

Comments
 (0)