Pytorch
文章平均质量分 67
天涯海阁未走远
这个作者很懒,什么都没留下…
展开
-
torch的CNN案例,mnist数据集下载缓慢的解决方案
今天在看莫凡的torch教程,cnn案例。封装的函数不一样,所以第一次跑demo的时候需要下载数据集。但是我发现,可能是由于网站的维护问题,是的数据集下载非常非常缓慢,难以忍受。所以在结合源码以及以前下载的matlab环境下使用的数据集。做了以下处理:数据集:另外先在目录下新建mnist文件夹,在mnist文件夹下再建raw和processed文件夹。自己的电脑只是路径的不...原创 2018-08-04 20:39:09 · 5463 阅读 · 4 评论 -
x.view(x.size(0), -1)的解释
pytorch教程中,用cnn实现mnist的分类。其中class中有一段代码:def forward(self, x): x = self.conv1(x) x = self.conv2(x) # t = x.size(0) 0->50,1->32,2->7,3->7 x = x.view(x.size(0), -1) ...原创 2018-08-04 21:11:53 · 37627 阅读 · 9 评论