vae

A PyTorch implementation of Auto-Encoding Variational Bayes

Installation

git clone https://github.com/andregaio/dcgan.git
cd vae
conda create -n vae python=3.8
conda activate vae
pip install -r requirements.txt

Dataset

MNIST

Models

• VAE

Training

python vae/train.py

Inference

python vae/infer.py --weights weights/last.pt --out test/image.png

Results

Notes

• Some has been borrowed from https://github.com/Jackson-Kang/Pytorch-VAE-tutorial/tree/master