Skip to content

Files

Latest commit

48a4b51 · Jan 19, 2017

History

History
This branch is 518 commits behind pytorch/examples:main.

vae

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
Jan 17, 2017
Jan 19, 2017
Jan 19, 2017

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