Параллелизуйте проверку одной эпохи с обучением другой в Pytorch Lightning. ⇐ Python

[Anonymous]

Я обучаю модель молнии Pytorch на графическом процессоре. Я проверяю его в каждую эпоху, и каждый запуск проверки занимает около 30 минут на наборе данных, достаточно большом, чтобы дать мне надежную оценку.
Я хотел бы ускорить процесс, выполняя параллельную проверку ( на ЦП) с обучением следующей эпохи. Поддерживает ли pytorch Lightning эту функцию?
Единственное решение, о котором я мог подумать, — это запустить параллельный поток, который будет ожидать завершения новой эпохи (с последующим сохранением весов в контрольной точке). файл).

Подробнее здесь: https://stackoverflow.com/questions/708 ... rch-lightn