PyTorch VAE

A Collection of Variational AutoEncoders (VAEs) implemented in PyTorch.

Requirements

• Python >= 3.5
• PyTorch >= 1.3
• Pytorch Lightning >= 0.5.3 (GitHub Repo)

Installation

$ git clone https://github.com/AntixK/PyTorch-VAE
$ cd PyTorch-VAE
$ pip install -r requirements.txt

Usage

$ cd PyTorch-VAE
$ python run.py -c configs/<config-file-name.yaml>

---

Model	Paper	Reconstruction	Samples
VAE	https://arxiv.org/abs/1312.6114
WAE - MMD (RBF Kernel)	https://arxiv.org/abs/1711.01558
WAE - MMD (IMQ Kernel)	https://arxiv.org/abs/1711.01558

TODO

• VanillaVAE
• Conditional VAE
• Gamma VAE
• Beta VAE
• DFC VAE
• InfoVAE (MMD-VAE)
• WAE-MMD
• AAE
• TwoStageVAE
• VAE-GAN
• HVAE (VAE with Vamp Prior)
• IWAE
• VLAE
• FactorVAE
• PixelVAE