欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

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

程序员文章站 2022-03-03 14:41:24
...

在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)

即可