From 2d6767b2816f2fd344bf84ce4c4dc8432bbc0d3e Mon Sep 17 00:00:00 2001 From: Houssam Zenati Date: Thu, 31 Aug 2017 10:50:50 +0800 Subject: [PATCH] Update train.py Tiny modifications just to see the summaries and images on tensorboard with the steps displayed --- gan/train.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/gan/train.py b/gan/train.py index e57ea48..d62ad2e 100644 --- a/gan/train.py +++ b/gan/train.py @@ -104,8 +104,8 @@ inp, _ = next(data_generator) (_, sum_dis) = sess.run((train_discriminator, summary_discriminator), feed_dict={input_placeholder: inp}) - logwriter.add_summary(sum_dis) + logwriter.add_summary(sum_dis, batch) (_, sum_gen) = sess.run((train_generator, summary_generator)) - logwriter.add_summary(sum_gen) + logwriter.add_summary(sum_gen, batch) batch += 1