在loss.backward()的时候出现Found dtype Long but expected。但是看了一下loss的类型,发现是flaot32没问题。看了各种解决方案,发现一般是loss函数的输入类型不一致会导致这个问题。
比如我用的是nn.MSELoss(),两个输入必须都转换为float才行。
Pyotrch常见bug:Found dtype Long but expected
最新推荐文章于 2024-07-27 14:59:35 发布
在loss.backward()的时候出现Found dtype Long but expected。但是看了一下loss的类型,发现是flaot32没问题。看了各种解决方案,发现一般是loss函数的输入类型不一致会导致这个问题。
比如我用的是nn.MSELoss(),两个输入必须都转换为float才行。