Skip to content
VAE models for experiments and education usage.
Branch: master
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
data
.gitignore
LICENSE
README.md
vae_keras.py
vae_tf.py

README.md

VAE

VAE models for experiments and education usage.

Repo Structure

data

Saving data of MNIST.

vae_tf.py

A python script implemented by tensorflow and experiments conducted.

vae_keras.py

A python script implemented by keras and experiments conducted.

Environment

  • python2
  • Tensorflow1.4.1
  • Keras2.1.5

Usage

  • the keras version:

python vae_keras.py

  • the tensorflow version:

python vae_tf.py

tensorboard --logdir=tf_record/

Experiment Results

keras version

epoch=5, distribution of z:

epoch=5, the output of generator(decoder):

epoch=50, distribution of z:

epoch=50, the output of generator(decoder):

tensorflow version

training for 1000 steps, and the training loss is as follows:

More Information

Please visit http://skyhigh233.com/blog/2018/04/04/vae/ for more information.

License

MIT LICENSE

You can’t perform that action at this time.