Merge pull request #843 from PaddlePaddle/log

fix log for u2

deepspeech/exps/u2/model.py

@@ -123,13 +123,13 @@ class U2Trainer(Trainer):
 
         iteration_time = time.time() - start
 
-        if (batch_index + 1) % train_conf.log_interval == 0:
-            for k, v in losses_np.items():
-                report(k, v)
-            report("batch_size", self.config.collator.batch_size)
-            report("accum", train_conf.accum_grad)
-            report("step_cost", iteration_time)
+        for k, v in losses_np.items():
+            report(k, v)
+        report("batch_size", self.config.collator.batch_size)
+        report("accum", train_conf.accum_grad)
+        report("step_cost", iteration_time)
 
+        if (batch_index + 1) % train_conf.accum_grad == 0:
             if dist.get_rank() == 0 and self.visualizer:
                 losses_np_v = losses_np.copy()
                 losses_np_v.update({"lr": self.lr_scheduler()})
@@ -223,7 +223,9 @@ class U2Trainer(Trainer):
                             msg += f"{v:>.8f}" if isinstance(v,
                                                              float) else f"{v}"
                             msg += ","
-                        logger.info(msg)
+                        if (batch_index + 1
+                            ) % self.config.training.log_interval == 0:
+                            logger.info(msg)
                         data_start_time = time.time()
                 except Exception as e:
                     logger.error(e)