先说一下,我之前没学过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