RuntimeError: bool value of Tensor with more than one value is ambiguous
Pytorch 跑模型报错。
if labels:
loss = self.loss_fn(out, labels)
return out, loss
else:
return out
if labels 那一行报错。
这一行代码想判断labels是否为空。
因为labels是torch.tensor类型,不能直接转换成bool值。
需要改成:
if labels is not None。
原创
2021-09-14 19:14:19 ·
832 阅读 ·
0 评论