Память графического процессора Pytorch продолжает увеличиваться с каждой партией ⇐ Python

[Anonymous]

Я обучаю модель CNN на изображениях. Изначально я тренировался на патчах изображений размером (256, 256) и всё было нормально. Затем я изменил свой загрузчик данных, чтобы загружать изображения в формате Full HD (1080, 1920), и после некоторой обработки я обрезал изображения. В этом случае объем памяти графического процессора продолжает увеличиваться с каждым пакетом. Почему это происходит?
PS: При отслеживании потерь я выполняю loss.detach().item(), чтобы потери не сохранялись на графике.< /п>

Подробнее здесь: https://stackoverflow.com/questions/706 ... very-batch