Skip to content
This repository has been archived by the owner on Mar 17, 2021. It is now read-only.

Commit

Permalink
Merge pull request #475 from javierbg/fix_tensorboard_gan
Browse files Browse the repository at this point in the history
Fixed bug when logging discriminator loss for tensorboard in GAN application
  • Loading branch information
wyli committed Jan 17, 2020
2 parents bde1ee4 + b3158ca commit 7421b01
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions niftynet/application/gan_application.py
Original file line number Diff line number Diff line change
Expand Up @@ -215,10 +215,10 @@ def switch_sampler(for_training):
collection=CONSOLE)
# variables to display in tensorboard
outputs_collector.add_to_collection(
var=lossG, name='lossG', average_over_devices=False,
var=lossD, name='lossD', average_over_devices=True,
collection=TF_SUMMARIES)
outputs_collector.add_to_collection(
var=lossG, name='lossD', average_over_devices=True,
var=lossG, name='lossG', average_over_devices=False,
collection=TF_SUMMARIES)

with tf.name_scope('Optimiser'):
Expand Down

0 comments on commit 7421b01

Please sign in to comment.