Update utils.py

2025-08-26 18:24:18 +00:00 · 2021-09-29 16:01:53 +08:00 · 2021-09-29 16:01:53 +08:00 · d04271219c
commit d04271219c
parent 28fa30a255
1 changed files with 2 additions and 5 deletions
--- a/icefall/utils.py
+++ b/icefall/utils.py
@ -478,10 +478,7 @@ 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: