Pytorch学习笔记
Mundane_World
这个作者很懒,什么都没留下…
展开
-
Pytorch读取图片
Pytorch的torchvision虽然包含了几个数据集的读取API,例如CAFRI10,Imagenet,MNIST等,但这远远不够。实际应用中,我们需要从各种不同的数据集中读取图片。 Pytorch自定义读取数据的方式,主要用到两个类: torch.utils.data.Dataset和torch.utils.data.DataLoader 为了自由读取数据集中的数据...原创 2018-07-01 11:42:21 · 10585 阅读 · 0 评论 -
Pytorch :Trying to backward through the graph a second time, but the buffers have already been freed
最近在学习Pytorch,刚用Pytorch重写了之前用Tensorlfow写的论文代码。 首次运行就碰到了一个bug: Pytorch - RuntimeError: Trying to backward through the graph a second time, but the buffers have already been freed. Specify retain_grap...原创 2018-07-13 22:05:42 · 9559 阅读 · 6 评论