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。

by the way

python的基本类型如list 如 test = []可以通过if test 判断是否为空。