mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-08-26 18:24:18 +00:00
Update utils.py
This commit is contained in:
parent
d04271219c
commit
d0986e9787
@ -478,7 +478,8 @@ 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:
|
||||
|
Loading…
x
Reference in New Issue
Block a user