insujeon/IWAE-torch

Variational and Importance Weigthed Autoencoders in PyTorch

VAE and IWAE in PyTorch

Implementations of Variational Autoencoder (VAE) and Importance Weigthed Autoencoders (IWAE) in PyTorch

Currently

• Factorized Gaussian encoder
• Factorized Gaussian decoder (with range limited variance)
• VAE and IWAE cost functions
• MNIST example

References

• https://arxiv.org/abs/1312.6114
• https://arxiv.org/abs/1509.00519