本文共 270 字,大约阅读时间需要 1 分钟。
pytorch实现目标检测网络时,可能会出现一下问题:
RuntimeError: cuda runtime error (59) : device-side assert triggered at /pytorch/aten/src/THC/generic/THCTensorMath.cu:16出现上述错误可能是因为在box_utils.py脚本中 conf = labels[best_truth_idx] + 1,从而使得所有的label都加上了1。解决方法删除 +1就OK了。
转载地址:http://uwmbb.baihongyu.com/