初识生成对抗网络(12)——利用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__<

猜你喜欢

转载自blog.csdn.net/didi_ya/article/details/125434359
今日推荐