feature/1124 Updated object detection getting started guide to add PL…

… test
airctic · Jul 21, 2022 · 34e8fde · 34e8fde
1 parent f5add83
commit 34e8fde
Showing 1 changed file with 21 additions and 3 deletions.
diff --git a/notebooks/getting_started_object_detection.ipynb b/notebooks/getting_started_object_detection.ipynb
@@ -802,6 +802,24 @@
     "trainer.fit(light_model, train_dl, valid_dl)"
    ]
   },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "#### Testing using Pytorch Lightning\n",
+    "For testing, it is recommended to use a separate test dataset that the model did not see during training but for demonstration purposes we'll re-use the validation dataset."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "trainer = pl.Trainer()\n",
+    "trainer.test(light_model, valid_dl)"
+   ]
+  },
   {
    "cell_type": "markdown",
    "metadata": {
@@ -961,7 +979,7 @@
    "provenance": []
   },
   "kernelspec": {
-   "display_name": "Python 3.8.5 ('icevision')",
+   "display_name": "Python 3.10.4 64-bit",
    "language": "python",
    "name": "python3"
   },
@@ -975,7 +993,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.8.5"
+   "version": "3.10.4"
   },
   "metadata": {
    "interpreter": {
@@ -1002,7 +1020,7 @@
   },
   "vscode": {
    "interpreter": {
-    "hash": "8de9f07afee82c69462511c5dd73ef92dc31ce4377be5cdd30293ad211c0627b"
+    "hash": "916dbcbb3f70747c44a77c7bcd40155683ae19c65e1c03b4aa3499c5328201f1"
    }
   },
   "widgets": {