使用torch.nn.CrossEntroyLoss()出错 - pytorch吧

level 1

-紬- 楼主

图1的11行使用了该函数，但是出现报错，内容如图2，说应该用long类型；
所以我在图1第8行加了转换为long类型的代码，并且把y类型打印出来（如图3），发现在循环中前面一部分运行没有错误，但是后面某一步出错了，报错内容如图2，但是所打印出来的y类型都是long.
请教一下应该怎么修改，才能使torch.nn.CrossEntroyLoss函数运行成功

2023年06月16日 04点06分 1

level 1

-紬- 楼主

顶

2023年06月16日 05点06分 2

level 11

sealed_ss

报的错似乎不是那一行，是你eval下面那些

2023年06月17日 07点06分 3

-紬-

哈哈，我知道了，确实是这里的问题，谢谢

2023年06月20日 12点06分