pytorch下实现ssd目标检测算法运行时遇到的错误

pytorch下实现ssd目标检测算法运行时遇到的错误

代码地址：https://github.com/amdegroot/ssd.pytorch
版本：pytorch1.0

运行test.py遇到错误 ValueError: not enough values to unpack (expected 2, got 0)
如下图：

解决：
修改 detection.py of的line 49如下：
原来：
if scores.dim() == 0:
continue
修改为：
if scores.size(0) == 0:
continue

参考：https://github.com/amdegroot/ssd.pytorch/issues/154