Awesome

dragan-pytorch

PyTorch implementation of DRAGAN (https://arxiv.org/abs/1705.07215)

Code based on the original implementation by the authors.

The following repositories were also used as a reference on how to implement the gradient penalty in PyTorch:

• https://github.com/t-vi/pytorch-tvmisc
• https://github.com/caogang/wgan-gp