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