Skip to content

Latest commit

 

History

History
25 lines (18 loc) · 545 Bytes

README.md

File metadata and controls

25 lines (18 loc) · 545 Bytes

generative-tf

Generative Models with TensorFlow

Variational Autoencoder

Kingma, D. P. & Welling, M. Auto-Encoding Variational Bayes

python generative-tf/train_mnist_vae.py \
     --epochs 5000 \
     --print-every-N 100 \
     --latent-dim 10 \
     --hidden-dim 500 \
     --batch-size 100 \
     --optimizer rmsprop

Burda, Y., Grosse, R. & Salakhutdinov, R. Importance Weighted Autoencoders. 1–12 (2015). at http://arxiv.org/abs/1509.00519

Add importance weighting with:

--importance-weighting