PyTorch踩坑指南
更新torchvision后在终端输出该错误提示时
# /usr/local/lib/python3.6/dist-packages/torchvision/ops/poolers.py in setup_scales(self, features, image_shapes)
line161: lvl_min = -torch.log2(torch.tensor(scales[0], dtype=torch.float32)).item()
问题原因是由于版本的变更,某些方法进行了重构,因此需要更改.
原创
2021-03-24 14:06:33 ·
92 阅读 ·
0 评论