diff --git a/paddlenlp/trainer/trainer.py b/paddlenlp/trainer/trainer.py index d65698f000cb..60403490ee0a 100644 --- a/paddlenlp/trainer/trainer.py +++ b/paddlenlp/trainer/trainer.py @@ -921,6 +921,7 @@ def _inner_training_loop( steps_trained_progress_bar.update(1) if steps_trained_in_current_epoch == 0: self._load_rng_state(resume_from_checkpoint) + self.timers and self.timers("read-data").start() continue elif steps_trained_progress_bar is not None: steps_trained_progress_bar.close()