[P] Implementation of Conditional WGAN and WGAN in pytorch • r/MachineLearning

This is our implementation of Conditional improved WGAN and improved WGAN in pytorch. Since this is our first-time working on GANs, it is harder than we thought. Although the reference code are already available (caogang-wgan in pytorch and improved wgan in tensorflow), the main part which is gan-64x64 is not yet implemented in pytorch. We realize that training GAN is really unstable. For instance, we stuck for one month and needed to test each component in our model to see if they are equivalent to their tf counterparts.