Variational Auto Encoders train.py - Train VAE vae_experiment.py - Experiments with VAE baseline_experiment.py - Experiments without VAE