Я хочу обучить модель обнаружения цели, используя CNN с pytorch, просто используя cnn, используемый в распознавании цифр Mnist, а размерный вывод равен 5, включая belif, left connner x, y и длину и ширину поля.
Я знаю, что это не очень хорошая идея для обнаружения, но я хочу, чтобы он хотя бы работал.
Теперь это ошибка, с которой я сталкиваюсь при обучении, а точнее, когда loss.backward (), ошибка: ошибка Cuda из памяти
введите описание изображения здесь
введите описание изображения здесь