This repository is for sharing the scripts of VAE.
You can train MNIST with any number of latent variable dimensions, and visualize the latent space from multiple perspectives.
- Python: 3.7
- fire: 0.4
You can use the following command to train and visualize VAE.
python main.py main --z_dim [The dimensions of latent variables]
You can start tensorboard with the following command. Note that The default URL is http://localhost:6006/.
tensorboard --logdir ./logs
The configuration of VAE is shown as below.
You can get the following images.
