出現這個問題就是tensorflow第二次使用的是第一次的就的變量,所以在第一次使用完成后,重置tensorflow的圖表就好了,在兩次使用中間加入:
tf.reset_default_graph()#重置tensorflow的舊變量,重置圖標
gan = GAN(gan_type='gan', data=data, batch_size=64, noise_dim=10, clip_num=0, critic_iter=5) gan.train(100) tf.reset_default_graph() gan = GAN(gan_type='wgan', data=data, batch_size=64, noise_dim=10, clip_num=0.1, critic_iter=5) gan.train(100)
加入tf.reset_default_graph()
程序就不再報錯