GAN and WGAN-GP

GAN and WGAN-gp implemented with pytorch and test on the mnist dataset

Installation

Clone this repo:

  git clone https://github.com/ZDDWLIG/WGAN-GP.git
  cd path/to/GAN-pytorch-master

Use conda to manage the python environment:

  conda create -n GAN python=3.8
  conda activate GAN
  pip install -r requirements.txt

For example, if you want to train WGAN-gp , then run：

  python WGAN-GP.py

Name		Name	Last commit message	Last commit date
Latest commit History 28 Commits
dataset		dataset
image		image
GAN.py		GAN.py
README.md		README.md
WGAN-GP.py		WGAN-GP.py
module.py		module.py
requirements.txt		requirements.txt