WGAN with gradient penalty