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
·
解决: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)
更多推荐
已为社区贡献2条内容
所有评论(0)