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
ec3b1bb015
commit
7faab70ddf
@ -478,8 +478,12 @@ class LossRecord(collections.defaultdict):
|
|||||||
for k, v in zip(keys, s.cpu().tolist()):
|
for k, v in zip(keys, s.cpu().tolist()):
|
||||||
self[k] = v
|
self[k] = v
|
||||||
|
|
||||||
def write_summary(self, tb_writer: SummaryWriter,
|
def write_summary(
|
||||||
prefix: str, batch_idx: int) -> None:
|
self,
|
||||||
|
tb_writer: SummaryWriter,
|
||||||
|
prefix: str,
|
||||||
|
batch_idx: int
|
||||||
|
) -> None:
|
||||||
"""Add logging information to a TensorBoard writer.
|
"""Add logging information to a TensorBoard writer.
|
||||||
|
|
||||||
Args:
|
Args:
|
||||||
|
Loading…
x
Reference in New Issue
Block a user