How to implement gradient penalty in PyTorch

[tom (Thomas V)]

Ajay and I discussed that a bit a while ago and there is a link to a blog post and Jupyter notebook doing the toy examples from the improved training article in pytorch:

Wasserstein loss layer/criterion

Hi @AjayTalati, thanks for the pointers! I will definitely look into implementing an application or two. In the meantime, I jotted down a few thoughts regarding the Improved Training paper for your amusement while we’re waiting for the grad of grad to be merged. Or maybe we find something to do with the original WGAN code, too. You have good holidays, best regards Thomas

Best regards

Thomas