您当前的位置: 首页 >  网络

wendy_ya

暂无认证

  • 1浏览

    0关注

    342博文

    0收益

  • 0浏览

    0点赞

    0打赏

    0留言

私信
关注
热门博文

基于PyTorch的生成对抗网络实战(5)——利用Pytorch搭建WGAN生成手写数字

wendy_ya 发布时间:2022-06-23 19:51:33 ,浏览量:1

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

〇、WGAN介绍

关于WGAN的详细介绍可以参考文章:GAN生成对抗网络入门介绍及DCGAN、WGAN等介绍。这里只介绍一下WGAN在代码方面与GAN或DCGAN的区别。 在论文《Wasserstein GAN》中,作者推了一大堆公式定理,最终给出了改进的算法实现流程,而改进后相比原始GAN的算法实现流程却只改了四点:

  • 判别器最后一层去掉sigmoid;
  • 生成器和判别器的loss不取log;
  • 每次更新判别器的参数之后把它们的绝对值截断到不超过一个固定常数c;
  • 不要用基于动量的优化算法(包括momentum和Adam),推荐RMSProp,SGD也行; 算法如下图所示: 在这里插入图片描述

WGAN

关注
打赏
1659256378
查看更多评论
立即登录/注册

微信扫码登录

0.0447s