readme: ---
  license: apache-2.0
  ---
  
  ### 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.
  
  
  