https://github.com/tsvikas/variational-autoencoder

variational autoencoder
https://github.com/tsvikas/variational-autoencoder

Last synced: about 2 months ago
JSON representation

variational autoencoder

• Host: GitHub
• URL: https://github.com/tsvikas/variational-autoencoder
• Owner: tsvikas
• Created: 2023-05-24T10:13:40.000Z (almost 2 years ago)
• Default Branch: master
• Last Pushed: 2024-12-16T19:46:43.000Z (5 months ago)
• Last Synced: 2024-12-16T20:41:41.136Z (5 months ago)
• Language: Python
• Size: 1.41 MB
• Stars: 0
• Watchers: 1
• Forks: 0
• Open Issues: 2
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

README

        
# variational-autoencoder

this is an implementation of [variational autoencoder](https://en.wikipedia.org/wiki/Variational_autoencoder) using pytorch-lightning.

there are some helper classes for a lightning datamodule (with support for tran-validation split) and lightning model.

after training with a 2d latent space, the latent space looks like this:

![image of the latent space](output/latent_space_samples.png)