/VariationalAutoencoder

A PyTorch implementation of a Variational Auto-encoder class and loss Function

Primary LanguagePython

Variational Autoencoder

Basic implementaion of a Variational Autoencoder on GPU using PyTorch. Can be used as a starting point for creating your own.

see "Auto-Encoding Variational Bayes": https://arxiv.org/pdf/1312.6114.pdf