Pytorch SSD jupyter notebook demo 报错:need more than 0 values to unpack

pytorch ssd github : https://github.com/amdegroot/ssd.pytorch
运行demo中的demo.ipynb时,出现 need more than 0 values to unpack的错误。

解决方法:
将ssd.pytorch/layers/functions/detection.py第 49 行修改为:

                # if scores.dim() == 0:
                if scores.size(0) == 0:
                    continue

然后需要重启jupyter notebook 重新运行。即可成功。

你可能感兴趣的:(python,pytorch)