vae
PyTorch Lightning implementation of Variational AutoEncoder (VAE)
This repository contains PyTorch Lightning implementation of VAE for encoding of visual observations with unsupervised learning. For more technical details, check out the papers listed in references.
Requirements
References
- Auto-Encoding Variational Bayes, Diederik P Kingma and Max Welling, 2014
- Simple and Effective VAE Training with Calibrated Decoders, Oleh Rybkin, Kostas Daniilidis, Sergey Levine, 2020