Skip to content

Latest commit

 

History

History
23 lines (17 loc) · 595 Bytes

README.md

File metadata and controls

23 lines (17 loc) · 595 Bytes

Auto-Encoding Variational Bayes

This repo. contains a reimplementation of the variational auto encoder based on the original paper "Auto-Encoding Variational Bayes". by Kingma et.al
Paper Link: https://arxiv.org/abs/1312.6114

Clone the repository

git clone https://github.com/FaisalAhmed0/variational-autoencoder

you can setup a new environment and install requirements.txt

conda create -n vae_env 
pip3 install -r requirements.txt 

activate the new environment and run train.py

conda activate vae_env
python train.py --dataset mnist