Kernel died error while running the colab

I tried a smaller model and this still fails in the Generate Predictions On Test Data phase. The logs do not point to any particular code failure.