Skip to content

danielkurniadi/Numpy-Variational-Autoencoder

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

21 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Variational Autoencoder in NumPy

This is a full implementation of a simple VAE written entirely in Numpy (and Cupy). The code runs very slow on CPU so using a GPU with Cupy is recommended.

Original Paper: Auto-Encoding Variational Bayes, Diederik P Kingma, Max Welling

Used preprocessing code and inspired by @shinseung428's Vanilla GAN.

Implementation Details

  • Learns the MNIST Dataset
  • Xavier Initialization
  • Adam Optimizer

Requirements

  • Numpy
  • PIL (visualize results)
  • Cupy (Optional)

Network

network

Author

IqDF / @iqDF

About

Variational Autoencoder module simply build using Numpy and Numpy only

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Contributors 2

  •  
  •  

Languages