pytorch.generative.learning
This repository is for learning generative models using PyTorch.
Requirements
- Python >= 3.6
- PyTorch == 0.3.1
- torchvision == 0.2
- numpy
- tqdm
- visdom
Contents
gans
GANs -
DCGAN (Deep Convolutional Generative Adversarial Network):
python dcgan.py
-
WGAN (Wasserstein GAN)
python wgan.py