/autoencoders

Code of some simple autoencoders I'm learning about

Primary LanguagePython

Autoencoders

In this repo I'll keep the code of some tutorial autoencoders.

Content:

  • convolutional_vae.py: A Convolutional Variational Autoencoder adapted from the Tensorflow 2 Tutorial. It works for any image shape that is n x n x k with n power of 2.
  • train_mnist.py: Code to run the training On the MNIST dataset