/WGAN-GP-PyTorch

A python3 implementation of WGAN-GP

Primary LanguagePython

WGAN-GP

A python3&torch1.+ implementation of WGAN-GP.

Prerequisites

  • Python3.6+
  • torch 1.+
  • numpy
  • scipy
  • matplotlib
  • A nvidia GPU

Acknowledge

Based on the code of [caogang/wgan-gp].

(Motivation: when I use the code of [caogang/wgan-gp], I got stuck at the data loading stage. Hence I build this version of WGAN-GP, which works fine under my environment.