出现 Expected object of scalar type Long but got scalar type Int for argument #2 报错

        当我运行代码时,出现了标题样式的报错,经过网上的搜索,最后发现是损失函数CrossEntropyLoss()的问题。问题是出现在我的输入类型不同,因为该损失函数的输入类型应该是long型的所以修改方法如下图所示

 如图所示我只要把我的labels变量改成long型之后,程序就可以继续运行。

猜你喜欢

转载自blog.csdn.net/qq_45710342/article/details/121692260