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

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