PyTorch 提示 RuntimeError: expected scalar type Long but found Float

对于多分类一般采用的交叉熵损失函数 torch.nn.CrossEntropyLoss(),此时 label 的取值为 LongTensor 没有问题。

但在回归时,label 的取值为 FloatTensor,此时再使用 torch.nn.CrossEntropyLoss() 就会提示标题错误。

参考:

  1. RuntimeError: expected scalar type Long but found Float
  2. RuntimeError: expected scalar type Long but found Float (Pytorch)

PyTorch 提示 RuntimeError: expected scalar type Long but found Float
https://wonderhoi.com/2024/09/24/PyTorch-提示-RuntimeError-expected-scalar-type-Long-but-found-Float/
作者
wonderhoi
发布于
2024年9月24日
许可协议