From 812eeecaa2784dc2af3c68ceed034ec99943118a Mon Sep 17 00:00:00 2001 From: wangguanzhong Date: Mon, 21 Mar 2022 11:22:43 +0800 Subject: [PATCH] fix step setting in vdl add_scalar (#5411) --- ppdet/engine/callbacks.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ppdet/engine/callbacks.py b/ppdet/engine/callbacks.py index e45e91d65..77ca94602 100644 --- a/ppdet/engine/callbacks.py +++ b/ppdet/engine/callbacks.py @@ -261,7 +261,7 @@ class VisualDLWriter(Callback): for loss_name, loss_value in training_staus.get().items(): self.vdl_writer.add_scalar(loss_name, loss_value, self.vdl_loss_step) - self.vdl_loss_step += 1 + self.vdl_loss_step += 1 elif mode == 'test': ori_image = status['original_image'] result_image = status['result_image'] -- GitLab