Pyotrch常见bug：Found dtype Long but expected

最新推荐文章于 2025-08-04 19:07:53 发布

飞鱼~

最新推荐文章于 2025-08-04 19:07:53 发布

阅读量1k

点赞数

CC 4.0 BY-SA版权

分类专栏： pytorch 文章标签： bug python 人工智能

本文链接：https://blog.youkuaiyun.com/xinong123456123/article/details/128449462

pytorch 专栏收录该内容

2 篇文章

订阅专栏

本文探讨了在PyTorch中使用loss.backward()时遇到的类型错误问题，通常是因为loss函数输入的数据类型不一致导致。以nn.MSELoss为例，详细解释了如何确保所有输入均为float类型来规避此问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在loss.backward()的时候出现Found dtype Long but expected。但是看了一下loss的类型，发现是flaot32没问题。看了各种解决方案，发现一般是loss函数的输入类型不一致会导致这个问题。
比如我用的是nn.MSELoss()，两个输入必须都转换为float才行。