BicycleGAN icon indicating copy to clipboard operation
BicycleGAN copied to clipboard

Whether it is right about KL_loss

Open yangzhikai opened this issue 5 years ago • 1 comments

I find the calculation of KL-loss is different from VAE. Because: latent_loss = 0.5 * tf.reduce_sum( tf.square(z_mean) + tf.square(z_stddev)                       - tf.log(tf.square(z_stddev)) - 1, 1 ) I think the "self.mu.pow(2) " of line-154 in bicycle_gan_model.py should be "self.mu.pow(-2)".

yangzhikai avatar May 20 '19 15:05 yangzhikai

@pathak22 I think it should be pow(2) according to Equation (10) in the VAE paper.

junyanz avatar May 21 '19 02:05 junyanz