基于PyTorch的生成对抗网络实战（6）——利用Pytorch搭建WGAN-GP生成手写数字

• • 〇、WGAN-GP介绍
  • 一、生成器
  • 二、判别器
  • 三、迭代训练过程
  • 四、结果演示
  • 五、完整代码

WGAN-GP (Wasserstein GAN using gradient penalty)，即在WGAN的基础上加入gradient penalty梯度惩罚项，可参考论文：《Improved Training of Wasserstein GANs》。

生成器类与WGAN相同，代码如下：

class Generator(nn.Module):
    def __init__<