alihan/TensorFlow-2.x-Tutorials
1
0
Fork 0
mirror of https://github.com/dragen1860/TensorFlow-2.x-Tutorials.git synced 2021-05-12 18:32:23 +03:00
Code Issues Projects Releases Wiki Activity

Update wgan_train.py

Browse Source 
enable training for gp
This commit is contained in:
kelvinkoh0308
2019-09-18 18:27:57 +09:00
committed by GitHub
parent 439e0a1bfa
commit 9e92b2d994
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -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()