首页 > 编程语言 >(Pytorch第三天)复制官网教程程序遇到的一系列报错

(Pytorch第三天)复制官网教程程序遇到的一系列报错

时间:2023-02-17 21:57:10浏览次数:42  
标签:教程 tensor Pytorch 报错 cuda 官网 size

先说一下,我之前没学过Python,但是鉴于我是一名c语言高手(雾),我决定现学现用。

首先是https://pytorch.org/tutorials/beginner/basics/optimization_tutorial.html的

 

 位置,由于按照前面教程把model放cuda上了,但是数据还在cpu上,所以我加了

 

 device = “cuda”

但是我不能一开始就把所有数据放cuda上吗?(哪样更合理呢)

另外就是

 

这个位置报错The size of tensor a (64) must match the size of tensor b (10) at non-singleton dimension 1,好像说两个tensor的维度不同?

 

 原来是我把数据标签转成onehot了。。。。

这个事情告诉我们贴代码就老老实实贴,不要自我发挥(雾)

 

标签:教程,tensor,Pytorch,报错,cuda,官网,size
From: https://www.cnblogs.com/IrregularRhythm/p/17131574.html

相关文章