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

Logo

为开发者提供学习成长、分享交流、生态实践、资源工具等服务,帮助开发者快速成长。

更多推荐