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	Link
WAE - MMD (RBF Kernel)	Link
WAE - MMD (IMQ Kernel)	Link
Beta-VAE	Link	![][8]	![][7]
Disentangled Beta-VAE	Link	![][10]	![][9]

TODO

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