当我运行代码时,出现了标题样式的报错,经过网上的搜索,最后发现是损失函数CrossEntropyLoss()的问题。问题是出现在我的输入类型不同,因为该损失函数的输入类型应该是long型的所以修改方法如下图所示
如图所示我只要把我的labels变量改成long型之后,程序就可以继续运行。
版权声明:本文为qq_45710342原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。
当我运行代码时,出现了标题样式的报错,经过网上的搜索,最后发现是损失函数CrossEntropyLoss()的问题。问题是出现在我的输入类型不同,因为该损失函数的输入类型应该是long型的所以修改方法如下图所示
如图所示我只要把我的labels变量改成long型之后,程序就可以继续运行。