while training my model google colab stops working and kernel got disconnected. What can I do to restart and continue my training ?