У меня есть простая модель автоэнкодера: 
Теперь я хотел бы обучить этому на GTX 1080 с 8 ГБ памяти, но я всегда выхожу из ошибки памяти (OOM):
Однако одна партия данных меньше 10 МБ:
Итак, почему я получаю ошибки OOM? Общее количество параметров составляет 291 407 612, что должно составлять ~ 1 ГБ ОЗУ. Все промежуточные результаты можно рассчитать по частям, поэтому пиковая память не должна превышать 8 ГБ, верно?