未验证 提交 41036408 编写于 作者: W Walter 提交者: GitHub

Merge pull request #1186 from RainFrost1/trainer

fix loss_msg bugs
......@@ -34,7 +34,7 @@ def update_loss(trainer, loss_dict, batch_size):
for key in loss_dict:
if key not in trainer.output_info:
trainer.output_info[key] = AverageMeter(key, '7.5f')
trainer.output_info[key].update(loss_dict[key].numpy()[0], batch_size)
trainer.output_info[key].update(loss_dict[key].numpy()[0], batch_size)
def log_info(trainer, batch_size, epoch_id, iter_id):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册