From 7faab70ddf10acd8bb9882ac765c88308a7fa215 Mon Sep 17 00:00:00 2001
From: Mingshuang Luo <37799481+luomingshuang@users.noreply.github.com>
Date: Wed, 29 Sep 2021 16:43:01 +0800
Subject: [PATCH] Update utils.py

---
 icefall/utils.py | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/icefall/utils.py b/icefall/utils.py
index 3e84fcc6a..09ae6d7c0 100644
--- a/icefall/utils.py
+++ b/icefall/utils.py
@@ -478,8 +478,12 @@ class LossRecord(collections.defaultdict):
         for k, v in zip(keys, s.cpu().tolist()):
             self[k] = v
 
-    def write_summary(self, tb_writer: SummaryWriter, 
-        prefix: str, batch_idx: int) -> None:
+    def write_summary(
+        self,
+        tb_writer: SummaryWriter,
+        prefix: str,
+        batch_idx: int
+    ) -> None:
         """Add logging information to a TensorBoard writer.
         
         Args: