just do it on colab, save checkpoints, and keep training