From 0ee4ee56e5971315e4779bef87c9ff7220001db3 Mon Sep 17 00:00:00 2001 From: huangyuxin Date: Mon, 27 Sep 2021 06:53:57 +0000 Subject: [PATCH] optimize the log --- deepspeech/exps/u2/model.py | 1 + deepspeech/training/trainer.py | 1 + 2 files changed, 2 insertions(+) diff --git a/deepspeech/exps/u2/model.py b/deepspeech/exps/u2/model.py index 5cb0962a7..1afd9b101 100644 --- a/deepspeech/exps/u2/model.py +++ b/deepspeech/exps/u2/model.py @@ -216,6 +216,7 @@ class U2Trainer(Trainer): msg += f"{v:>.8f}" if isinstance(v, float) else f"{v}" msg += "," + msg = msg[:-1] # remove the last "," if (batch_index + 1 ) % self.config.training.log_interval == 0: logger.info(msg) diff --git a/deepspeech/training/trainer.py b/deepspeech/training/trainer.py index 15259f0eb..35b1690b8 100644 --- a/deepspeech/training/trainer.py +++ b/deepspeech/training/trainer.py @@ -263,6 +263,7 @@ class Trainer(): msg += f"{v:>.8f}" if isinstance(v, float) else f"{v}" msg += "," + msg = msg[:-1] # remove the last "," logger.info(msg) data_start_time = time.time() except Exception as e: