首页 » 机器视觉 » 正文

RuntimeError: Boolean value of Tensor with more than one value is ambiguous

2023-04-11 483 0

文章目录[隐藏]

解决：RuntimeError: Boolean value of Tensor with more than one value is ambiguous问题

深度学习训练模型报错：如上。
原因：

loss = loss_func(out, label)

解决方案一：criteria = nn.CrossEntropyLoss()
loss = criteria(out, label)
解决方案二：loss = torch.nn.functional.cross_entropy(out, label)

版权声明：本文为CSDN博主「糖尛果」的原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接及本声明。
原文链接：https://blog.csdn.net/m13526413031/article/details/116426016

我还没有学会写个人说明！

要发表评论，您必须先登录。