当前位置:网站首页>初识生成对抗网络(12)——利用Pytorch搭建WGAN-GP生成手写数字

初识生成对抗网络(12)——利用Pytorch搭建WGAN-GP生成手写数字

2022-06-25 07:02:00 wendy_ya

〇、WGAN-GP介绍

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

一、生成器

生成器类与WGAN相同,代码如下:

class Generator(nn.Module):
    def __init__<
原网站

版权声明
本文为[wendy_ya]所创,转载请带上原文链接,感谢
https://wendy.blog.csdn.net/article/details/125434359