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.