Merge pull request dragen1860#35 from kelvinkoh0308/patch-11

Update wgan_train.py

Author: dragen1860

深度学习与TensorFlow入门实战-源码和PPT/lesson51-WGAN实战/wgan_train.py

@@ -69,7 +69,7 @@ def gradient_penalty(discriminator, batch_x, fake_image):
 
     with tf.GradientTape() as tape:
         tape.watch([interplate])
-        d_interplote_logits = discriminator(interplate)
+        d_interplote_logits = discriminator(interplate,is_training)
     grads = tape.gradient(d_interplote_logits, interplate)
 
     # grads:[b, h, w, c] => [b, -1]
@@ -170,4 +170,4 @@ def main():
 
 
 if __name__ == '__main__':
-    main()
+    main()