Sitelet https://github.com/sarus-tech/tf2-published-models/tree/master/vae

Name		Name	Last commit message	Last commit date
parent directory ..
Pipfile		Pipfile
model.py		model.py
readme.md		readme.md
train.py		train.py
utils.py		utils.py

readme.md

Variational Autoencoder

Tensorflow 2 generic implementation of a variational autoencoder (VAE).

Usage

To train the model on MNIST use

python train.py

To see all training options use

python train.py -h