I’ve fixed the WassersteinGAN code via https://github.com/martinarjovsky/WassersteinGAN/commit/e553093d3b2a44a1b6d0c8739a973598af6aa535 161
@apaszke in my (old code) case I’ve hacked up reusing variables carefully, but it’s a hack.