Ecosyste.ms: Awesome
An open API service indexing awesome lists of open source software.
https://github.com/soooda/vae
My reproduction of the base VAE using Pytorch
https://github.com/soooda/vae
deep-learning generative-model python pytorch
Last synced: 9 days ago
JSON representation
My reproduction of the base VAE using Pytorch
- Host: GitHub
- URL: https://github.com/soooda/vae
- Owner: Soooda
- Created: 2023-12-22T03:43:34.000Z (11 months ago)
- Default Branch: master
- Last Pushed: 2024-01-27T16:23:20.000Z (10 months ago)
- Last Synced: 2024-01-27T17:29:56.480Z (10 months ago)
- Topics: deep-learning, generative-model, python, pytorch
- Language: Python
- Homepage:
- Size: 6.7 MB
- Stars: 0
- Watchers: 2
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
# Variational Autoencoder
This repo contains my reproduction of the base VAE paper in Pytorch. This model is using **MNIST** dataset.
Run `python train.py -h` to find exact flags you can set to kick start the training.
Run `python evaulate.py -h` to find exact flags you can set to use the trained model to generate new data.
## References
* [Auto-Encoding Variational Bayes](https://arxiv.org/abs/1312.6114)