phillipi / pix2pix

Image-to-image translation with conditional adversarial nets
https://phillipi.github.io/pix2pix/
Other
10.14k stars 1.71k forks source link

WGANGP with pix2pix #121

Closed refu0523 closed 6 years ago

refu0523 commented 6 years ago

I've tried to combine improved WGAN with pix2pix, but I got a very unstable training even got exploded ( NaN). I have also implemented original WGAN with pix2pix, and it works well. It seems like the clipping way does make sure the weight not exploded, but the gradient penalty doesn't work well. Does anyone have any idea or experience to combine these two?

rohitsaluja22 commented 6 years ago

Hi is it possible for you to share the code of original WGAN with pix2pix, and improved WGAN with pix2pix?