Pytorch SSD jupyter notebook demoエラー:need more than 0 values to unpack

372 ワード

pytorch ssd github : https://github.com/amdegroot/ssd.pytorchdemoの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を再起動して再実行する必要があります.が表示されます.