frederikschubert / variational-autoencoder

An example implementation of the Variational Autoencoder

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Variational Autoencoder

This repository contains some experimental implementations of the Variational Autoencoder.

Installation

conda env create -f environment.yml

Usage

# Train a VAE on mnist using fully connected neural networks as encoder/decoder
python vae.py

# Train a VAE on mnist using convolutional neural networks as encoder/decoder
python vae.py with cnn

# Train a conditional VAE
python vae.py with conditional

# Train a convolutional VAE on mnist with batch size 128
python vae.py with cnn "batch_size=128"

About

An example implementation of the Variational Autoencoder


Languages

Language:Python 100.0%