VAE for color images in PyTorch Lightning

This repo is an implementation for the matching medium tutorial

reconstructions on cifar-10

To run

pip install -r requirements.txt

python vae.py --gpus 1