Update utils.py

This commit is contained in:
Mingshuang Luo 2021-09-29 14:13:31 +08:00 committed by GitHub
parent 8d07ce2185
commit 28fa30a255
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -486,8 +486,8 @@ class LossRecord(collections.defaultdict):
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():