Wasserstein GAN with gradient penalty