Where-are-they-looking-PyTorch 代码Error: Bool value of Tensor with more than one value is ambiguous

在Where-are-they-looking-PyTorch中的utils.py中有一个判断

           if temp < best:
                best = temp

报出

Error: Bool value of Tensor with more than one value is ambiguous

解决方法
把getdata.py中的predict函数含返回值改成

return hm_base.view(-1, 227 * 227)

即可

你可能感兴趣的:(Python相关学习,机器学习)