jaanli / variational-autoencoder

Variational autoencoder implemented in tensorflow and pytorch (including inverse autoregressive flow)
https://jaan.io/what-is-variational-autoencoder-vae-tutorial/
MIT License
1.15k stars 256 forks source link

Compatibility with TF v1.2 #3

Closed eitanrich closed 7 years ago

eitanrich commented 7 years ago

Great tutorial! Thanks.

To run with TensorFlow version 1.2, I had to change:

kl = tf.reduce_sum(distributions.kl(q_z.distribution, p_z), 1)

to

kl = tf.reduce_sum(distributions.kl_divergence(q_z.distribution, p_z), 1)

Eitan

jaanli commented 7 years ago

Thanks Eitan!! Fixed :)