From 7d4d1bc26d52cd87eb97536ff154c7f127b68a55 Mon Sep 17 00:00:00 2001 From: Konrad Heidler Date: Wed, 10 Jul 2019 04:32:39 +0200 Subject: [PATCH] Two fixes for info_gan.md example Code (#15323) * Two fixes for info_gan.md example Code * retrigger CI --- docs/tutorials/gluon/info_gan.md | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/docs/tutorials/gluon/info_gan.md b/docs/tutorials/gluon/info_gan.md index 91adf6c75108..c35df69fb6d3 100644 --- a/docs/tutorials/gluon/info_gan.md +++ b/docs/tutorials/gluon/info_gan.md @@ -339,9 +339,11 @@ with SummaryWriter(logdir='./logs/') as sw: fake_image = generator(g_input) sw.add_scalar(tag='Loss_D', value={'test':d_error_epoch.asscalar()/count}, global_step=counter) - sw.add_scalar(tag='Loss_G', value={'test':d_error_epoch.asscalar()/count}, global_step=counter) + sw.add_scalar(tag='Loss_G', value={'test':g_error_epoch.asscalar()/count}, global_step=counter) sw.add_image(tag='data_image', image=((fake_image[0]+ 1.0) * 127.5).astype(np.uint8) , global_step=counter) sw.flush() + + counter += 1 discriminator.save_parameters("infogan_d_latest.params") generator.save_parameters("infogan_g_latest.params")