标签： detection pytorch
我想使用带有pytorch的CNN训练目标检测模型，仅使用Mnist数字识别中使用的cnn，并且输出的尺寸为5，包括belif，左connner x，y和框的长度和宽度。
我知道进行检测不是一个好主意，但我想至少使其运行。
现在这就是我在训练时遇到的错误，确切地说，是在loss.backward（）错误时出现的错误：Cuda错误内存不足
enter image description here
