diff --git a/ppgan/engine/trainer.py b/ppgan/engine/trainer.py
index 7a1e1a81a9c676de498092162805973f7b22dd09..4812a125ee2a4ec4e32d051d2d9be3fbaad8e928 100644
--- a/ppgan/engine/trainer.py
+++ b/ppgan/engine/trainer.py
@@ -160,6 +160,9 @@ class Trainer:
             batch_cost_averager.record(time.time() - step_start_time,
                                        num_samples=self.cfg.get(
                                            'batch_size', 1))
+
+            step_start_time = time.time()
+
             if self.current_iter % self.log_interval == 0:
                 self.data_time = reader_cost_averager.get_average()
                 self.step_time = batch_cost_averager.get_average()
@@ -172,8 +175,6 @@ class Trainer:
             if self.current_iter % self.visual_interval == 0:
                 self.visual('visual_train')
 
-            step_start_time = time.time()
-
             self.model.lr_scheduler.step()
 
             if self.by_epoch: