tch-rs icon indicating copy to clipboard operation
tch-rs copied to clipboard

Add WGAN-GP example

Open jerry73204 opened this issue 5 years ago • 1 comments

I'd like to present the Wasserstein GAN with gradient penalty example. It is slightly modified from the gan example. It removes batch norms in discriminator as the paper suggested, and replace them with instance norms.

I tested against the MNIST images. It works fine on single channel (grayscale) image. https://imgur.com/k5nQryZ

However, I found the image is bulrring if three-channel RGB images are provided. https://imgur.com/GuCVnEy

The code is almost ready. If anyone can tell if it is an implementation or fine-tuning issue, I could fix it soon and merge.

jerry73204 avatar Jun 30 '20 02:06 jerry73204

I tested WGAN-GP and it also generated blurry images. I found sources that experienced with this issue (1, 2).

I see we already have a Realtivistic DCGAN example. WGAN is just a plus. I not sure about the plan. Perhaps we can

  • Discard this PR
  • Accept this PR but without RGB blurry example
  • Accept this PR with RGB blurry example.

jerry73204 avatar Jan 14 '21 05:01 jerry73204

Closing old issue as no activity in a while.

LaurentMazare avatar Dec 02 '22 12:12 LaurentMazare