Fixed bug when logging discriminator loss for tensorboard in GAN appl…

…ication
NifTK · Jan 16, 2020 · b3158ca · b3158ca
1 parent bde1ee4
commit b3158ca
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/niftynet/application/gan_application.py b/niftynet/application/gan_application.py
@@ -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'):