/Multigrid-PyTorch

PyTorch implementation of the paper: Learning Generative ConvNets via Multi-grid Modeling and Sampling by Gao et al.

Primary LanguagePython

Learning Generative ConvNets via Multi-grid Modeling and Sampling

A PyTorch implementation of the paper: Learning Generative ConvNets via Multi-grid Modeling and Sampling

Getting Started

(1) Download the code as a zip file and extract it (2) Copy the training data into ./data/dataset_name (3) Run the code

python main.py --dataset_name dataset_name

Prerequisites

PyTorch(>0.1.12). You can install it here.

Authors

Prateek Malhotra
Ruiqi Gao

Resources

You can find the official tensorflow implementation here.