出现这个问题就是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()
程序就不再报错