Two fixes for info_gan.md example Code (#15323)

* Two fixes for info_gan.md example Code * retrigger CI
apache · Jul 10, 2019 · 7d4d1bc · 7d4d1bc
1 parent d82c89a
commit 7d4d1bc
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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")