dragan-pytorch

May 27, 2017 ยท View on GitHub

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