cpwan/VGrow

VGrow for deep generative learning

VGrow in PyTorch

PyTorch implementation of Deep Generative Learning via Variational Gradient Flow.

Prerequisites

Python 3.5+
PyTorch v0.4.1

Usage

To run VGrow on MNIST, FashionMNIST and CIFAR10, use the following cmd with default arguments
python main.py --divergence KL --dataset <data_set> --dataroot <data_root>