Skip to content

Latest commit

 

History

History
9 lines (6 loc) · 613 Bytes

README.md

File metadata and controls

9 lines (6 loc) · 613 Bytes

Vanilla VAE Implementation

Make sure you have all the packages in requirements.txt installed. You may use the pip3 install -r requirements.txt command.

After installation, run the run_vae.py file. Inside the file you can find the necessary configuration parameters for the VAE. Setting the training flag to 0, will trigger the evaluation and a plot of 200 samples is saved in the project directory. If the latent (z) dimension is 2, in addition, you can find a plot of the distributions of the real digits in the latent space.

If you decide to trainn your own VAE, it should only take about 5min.