Skip to content

Files

This branch is 455 commits behind pytorch/examples:main.

vae

Basic VAE Example

This is an improved implementation of the paper Stochastic Gradient VB and the Variational Auto-Encoder by Kingma and Welling. It uses ReLUs and the adam optimizer, instead of sigmoids and adagrad. These changes make the network converge much faster.

pip install -r requirements.txt
python main.py