A Collection of Variational AutoEncoders (VAEs) implemented in PyTorch.
- Python >= 3.5
- PyTorch >= 1.3
- Pytorch Lightning >= 0.5.3 (GitHub Repo)
$ git clone https://github.com/AntixK/PyTorch-VAE
$ cd PyTorch-VAE
$ pip install -r requirements.txt
$ 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] |
- 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