PyTorch 提示 RuntimeError: expected scalar type Long but found Float
对于多分类一般采用的交叉熵损失函数 torch.nn.CrossEntropyLoss()
,此时 label 的取值为 LongTensor 没有问题。
但在回归时,label 的取值为 FloatTensor,此时再使用 torch.nn.CrossEntropyLoss()
就会提示标题错误。
参考:
PyTorch 提示 RuntimeError: expected scalar type Long but found Float
https://wonderhoi.com/2024/09/24/PyTorch-提示-RuntimeError-expected-scalar-type-Long-but-found-Float/