/Variational-Autoencoder-pytorch

Implementation of a convolutional Variational-Autoencoder model in pytorch.

Primary LanguagePython

Variational Autoencoder

This repository contains a convolutional-VAE model implementation in pytorch and trained on CIFAR10 dataset.

How to train

python main.py --config=./configs/cifar10.json