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

解决: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

糖尛果

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

暂无评论

发表评论

相关推荐

yolov5训练数据集划分

yolov5训练数据集划分 按照默认8:1:1划分训练集,测试集,验证集。 txt文件出现在imageset文件夹。 import os import randomtrainval_pe