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 1VAE for color images
This repo is an implementation for the matching medium tutorial
pip install -r requirements.txt
python vae.py --gpus 1VAE for color images
Apache License 2.0