Handle failed training jobs in notebook 007

openvinotoolkit · Oct 6, 2022 · b9ca0bb · b9ca0bb
1 parent 755ffe0
commit b9ca0bb
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/notebooks/007_train_project.ipynb b/notebooks/007_train_project.ipynb
@@ -255,7 +255,16 @@
    "outputs": [],
    "source": [
     "model = model_client.get_model_for_job(job)\n",
-    "print(model.overview)"
+    "\n",
+    "if model is not None:\n",
+    "    print(model.overview)\n",
+    "else:\n",
+    "    print(\n",
+    "        f\"Job '{job.name}' completed with status '{job.status.state}: \"\n",
+    "        f\"{job.status.message}', but did not result in a trained model. Most likely \"\n",
+    "        f\"the model training has failed, you could try restarting the training to \"\n",
+    "        f\"see if the problem persists.\"\n",
+    "    )"
    ]
   },
   {