Хранение набора данных на процессорах, но обучение на графических процессорах с помощью Lightning ⇐ Python

[Anonymous]

У меня возникла небольшая проблема, когда загрузчик данных загружает наборы обучения и проверки в графический процессор, и ему не хватает оперативной памяти, что приводит к завершению работы программы. Мне рекомендовали настроить устройство на процессор и передавать пакеты на графический процессор только по одному во время обучения. Проблема в том, что код написан с использованием Lightning, и я не могу понять, как его переопределить.
Мой ускоритель настроен на графический процессор, так как я тренируюсь на нем. Кто-нибудь знает, что нужно настроить в трейнере, чтобы Lightning не сбрасывала все в графические процессоры с самого начала?

Подробнее здесь: https://stackoverflow.com/questions/791 ... -lightning