Домашки в УИИ делаю в Колабе, на моем ноуте видеокарта не позволяет. Причем Колаб использую бесплатный, бывает что модель на одной эпохе учить по 10 минут и Колаб периодически капчу выводит, проверяет не пора ли закрыть сессию.
Единственно, приходиться периодически сохранять веса модели во время длительного обучения, так как могут закрыть сессию.