Не забудьте скачать итоговую модель, иначе гугл колаб ее удалит через некоторое время как вы отключитесь

Если я верно понимаю, то можно в пункте 3.1. смонтировать свой Google-Drive и в п.5.1 указать, чтобы тренированная модель сохранялась сразу к нам в облако.