Downloads · 30 days
17
3% of all-time downloads
VAE
Souranil/VAE
VAE is a machine learning model from Souranil. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for transformers. The card lists the license as apache-2.0.
This is inspired from vae-playground. This is an example where we test out vae and convvae models with multiple datasets like MNIST, celeb-a and MNIST-Fashion datasets.
All-time downloads
532
Public
Repo size
479 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.bin479 MB · 100%
At a glance
- Library
- transformers
- License
- apache-2.0
- Access
- Public
- Created
- Mar 2, 2022
- Updated
- Feb 18, 2022
- SHA
- abc07b4b
Model card
From the Hugging Face model README
VAE with Pytorch-Lightning
This is inspired from vae-playground. This is an example where we test out vae and conv_vae models with multiple datasets like MNIST, celeb-a and MNIST-Fashion datasets.
This also comes with an example streamlit app & deployed at huggingface.
Model Training
You can train the VAE models by using train.py and editing the config.yaml file.
Hyperparameters to change are:
- model_type [vae|conv_vae]
- alpha
- hidden_dim
- dataset [celeba|mnist|fashion-mnist]
There are other configurations that can be changed if required like height, width, channels etc. It also contains the pytorch-lightning configs as well.
- Library
- transformers
- License
- apache-2.0
- Created
- Mar 2, 2022
- Updated
- Feb 18, 2022