Параллелизм моделей, CUDA не хватает памяти в Pytorch

#python #pytorch #gpu #out-of-memory Вопрос: Я пытаюсь построить модель автоэнкодера, где ввод/вывод-это изображения RGB размером 256 x 256. Я пытался обучить модель на 1 графическом процессоре с 12 ГБ памяти,…

Продолжить чтениеПараллелизм моделей, CUDA не хватает памяти в Pytorch