Colab is awesome but I've found that GPU memory allocation can be a bit unreliable at times. It's advertised as providing you a GPU but in reality I think you're sharing the GPU with other machines which can lead to out-of-memory issues fitting models that you don't run into using a dedicated GPU instance