missed something

ADDED DISCONNECT BACK

Train_Colab.ipynb

@@ -490,6 +490,9 @@
         "\n",
         "# Finish the training process\n",
         "process.wait()"
+        "if Disconnect_after_training :
+        "  time.sleep(40)
+        "  runtime.unassign()
       ]
     },
     {
@@ -579,4 +582,4 @@
   },
   "nbformat": 4,
   "nbformat_minor": 0
-}
+}