https://github.com/williamfalcon/pytorch-lightning-vae

VAE for color images
https://github.com/williamfalcon/pytorch-lightning-vae

Last synced: 11 months ago
JSON representation

VAE for color images

• Host: GitHub
• URL: https://github.com/williamfalcon/pytorch-lightning-vae
• Owner: williamFalcon
• License: apache-2.0
• Created: 2020-12-05T11:57:46.000Z (over 5 years ago)
• Default Branch: main
• Last Pushed: 2021-09-08T11:34:36.000Z (almost 5 years ago)
• Last Synced: 2025-05-08T15:43:02.131Z (about 1 year ago)
• Language: Python
• Size: 291 KB
• Stars: 210
• Watchers: 2
• Forks: 20
• Open Issues: 10
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

README

          
# VAE for color images in PyTorch Lightning

This repo is an implementation for the matching medium tutorial

## reconstructions on cifar-10

![image](https://github.com/williamFalcon/pytorch-lightning-vae/blob/main/images/cifar_10.png)

## To run

```bash

pip install -r requirements.txt

python vae.py --gpus 1

```