From 34e8fdede2a7e2ba138cd8183607f7bbd54bf2a8 Mon Sep 17 00:00:00 2001 From: brown Date: Thu, 21 Jul 2022 13:14:10 -0400 Subject: [PATCH] feature/1124 Updated object detection getting started guide to add PL test --- .../getting_started_object_detection.ipynb | 24 ++++++++++++++++--- 1 file changed, 21 insertions(+), 3 deletions(-) diff --git a/notebooks/getting_started_object_detection.ipynb b/notebooks/getting_started_object_detection.ipynb index 37a111157..5061ba449 100644 --- 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": {