/pytorch-lightning-vae

VAE for color images

Primary LanguagePythonApache License 2.0Apache-2.0

VAE for color images in PyTorch Lightning

This repo is an implementation for the matching medium tutorial

reconstructions on cifar-10

image

To run

pip install -r requirements.txt

python vae.py --gpus 1