tensorflow-generative-model-collections icon indicating copy to clipboard operation
tensorflow-generative-model-collections copied to clipboard

Gradient Penalty code error in WGAN_GP

Open jwc0906 opened this issue 5 years ago • 2 comments

at 113 line in WGAN_GP, I recommend changing the code

alpha = tf.random_uniform(shape=self.inputs.get_shape(), minval=0.,maxval=1.) to alpha = tf.random_uniform(shape=[BATCH_SIZE,1,1,1], minval=0.,maxval=1.)

Because It must be created one alpha value for each batch

jwc0906 avatar Jul 08 '19 11:07 jwc0906

You are right. I have the same view.The performance improved after changing thie code.

AliceAria avatar Dec 02 '19 09:12 AliceAria

You are right. I have the same view.The performance improved after changing thie code.

I have a question that wheather the bn in Discrimator should be removed?

iskangkang avatar Jun 16 '20 01:06 iskangkang