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

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

Last synced: 3 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 (almost 4 years ago)
• Default Branch: main
• Last Pushed: 2021-09-08T11:34:36.000Z (about 3 years ago)
• Last Synced: 2024-04-12T00:51:17.688Z (7 months ago)
• Language: Python
• Size: 291 KB
• Stars: 199
• Watchers: 3
• 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

```