From a0994ee58ab65e7eca1888f5ca9816489b4ddd5b Mon Sep 17 00:00:00 2001 From: Mingshuang Luo <37799481+luomingshuang@users.noreply.github.com> Date: Wed, 29 Sep 2021 16:48:00 +0800 Subject: [PATCH] Update utils.py --- icefall/utils.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/icefall/utils.py b/icefall/utils.py index 09ae6d7c0..cfb6b686e 100644 --- a/icefall/utils.py +++ b/icefall/utils.py @@ -482,14 +482,14 @@ class LossRecord(collections.defaultdict): self, tb_writer: SummaryWriter, prefix: str, - batch_idx: int + batch_idx: int, ) -> None: """Add logging information to a TensorBoard writer. - + Args: tb_writer: a TensorBoard writer - prefix: a prefix for the name of the loss, - e.g. "train/valid_", or "train/current_". + prefix: a prefix for the name of the loss, e.g. "train/valid_", + or "train/current_" batch_idx: The current batch index, used as the x-axis of the plot. """ for k, v in self.norm_items():