From d586ee7efcf380805e1e87c4702e4e68d8de335c Mon Sep 17 00:00:00 2001
From: GuoxiaWang <mingzilaochongtu@gmail.com>
Date: Sun, 24 Mar 2024 21:53:36 +0800
Subject: [PATCH] fix read-data timer when ignore_data_skip=False and
 skip_profile_timer=False

---
 paddlenlp/trainer/trainer.py | 1 +
 1 file changed, 1 insertion(+)

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()