摘要: 我最近在学使用Pytorch写GAN代码,发现有些代码在训练部分细节有略微不同,其中有的人用到了detach()函数截断梯度流,有的人没用detch(),取而代之的是在损失函数在反向传播过程中将backward(retain_graph=True),本文通过两个 gan 的代码,介绍它们的作用,并分 阅读全文
posted @ 2020-11-09 23:40 凌逆战 阅读(5192) 评论(5) 推荐(5) 编辑
摘要: pytorch对一下常用的公开数据集有很方便的API接口,但是当我们需要使用自己的数据集训练神经网络时,就需要自定义数据集,在pytorch中,提供了一些类,方便我们定义自己的数据集合 torch.utils.data.Dataset:所有继承他的子类都应该重写 __len()__ , __geti 阅读全文
posted @ 2020-11-09 12:16 凌逆战 阅读(3898) 评论(2) 推荐(7) 编辑