Ускорение обучения модели машинного обучения

Обучение модели идёт сутками, а результат нужен сегодня.
Что получаете. Разбор, где именно уходит время, ускорение обучения без потери качества, замер до и после на ваших же данных, запуск с продолжением после обрыва.
Последний такой замер: обучение ускорено в 8.9 раза.
Работаю с PyTorch и обучением с подкреплением, считаю на видеокарте.
От вас код, данные и доступ к машине, на которой это крутится.
Сначала беру ваш прогон и показываю замер, где теряется время, дальше решаете, делать ускорение или нет.

• Ключевые слова:
• gpu
• подкреплением
• нейросеть
• оптимизация
• pytorch
• обучения
• ускорение
• кода
• обучение
• cuda
• машинное