Встретил проблему на финальном этапе обучения на гпу: среда колаба / каггла падает из-за нехватки доступной памяти. Это происходит уже после просчета всех итераций при, насколько я понимаю, выгрузке модели с гпу. Какие могут быть способы обойти это, кроме уменьшения числа итераций или перехода на цпу?