解决：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
暂无评论