未验证 提交 812eeeca 编写于 作者: W wangguanzhong 提交者: GitHub

fix step setting in vdl add_scalar (#5411)

上级 674840fb
...@@ -261,7 +261,7 @@ class VisualDLWriter(Callback): ...@@ -261,7 +261,7 @@ class VisualDLWriter(Callback):
for loss_name, loss_value in training_staus.get().items(): for loss_name, loss_value in training_staus.get().items():
self.vdl_writer.add_scalar(loss_name, loss_value, self.vdl_writer.add_scalar(loss_name, loss_value,
self.vdl_loss_step) self.vdl_loss_step)
self.vdl_loss_step += 1 self.vdl_loss_step += 1
elif mode == 'test': elif mode == 'test':
ori_image = status['original_image'] ori_image = status['original_image']
result_image = status['result_image'] result_image = status['result_image']
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册