50 行 PyTorch 代码搞定 GAN

本文介绍了如何使用50行PyTorch代码实现GAN,包括配置PyTorch环境、数据生成、生成网络与判别网络的定义。在初始化网络时,对数据进行预处理,通过对比不同返回方式的效果,说明了数据增强对于提升生成器性能的重要性。实验结果显示,返回data+diffs的方式能生成更接近真实数据的样本,因为其更高的维度提供了更强的学习能力。
摘要由CSDN通过智能技术生成

最近,一篇非常火的报道,使用pytorch 加 50 行核心代码模拟 GAN  对抗神经网络,自己尝试走了一遍,并对源码提出自己的理解。原文链接如下 https://medium.com/@devnag/generative-adversarial-networks-gans-in-50-lines-of-code-pytorch-e81b79659e3f#.nr3akui9z 

code 地址 https://github.com/devnag/pytorch-generative-adversarial-networks

首先配置pytorch, 这个直接去github 上的pytorch官网按照教程去配,链接地址 https://github.com/pytorch/pytorch

之后把code 下下来,直接运行就可以了。


数据的产生,正样本也就是使用一个均值为4,标准差为1.25的高斯分布,

torch.Tensor(np.random.normal(mu, sigma, (1, n)))
负样本是随机数

torch.rand(m, n) 

两个网路的定义,一个是生成网络,将其参数打印出来如下

Generator (

  (map1): Linear (1 -> 50)
  (map2): Linear (50 -> 50)
  (map3): Linear (50 -> 1)
)

输入是(1L,1L)
</

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值