From bf9ce8e794a9ca8858eda1a9331402289fbacebf Mon Sep 17 00:00:00 2001
From: Elihei2 <eheidari@student.ethz.ch>
Date: Tue, 24 Sep 2024 15:40:44 +0200
Subject: [PATCH] added lines to the tutorial

---
 docs/notebooks/segger_tutorial.ipynb | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/docs/notebooks/segger_tutorial.ipynb b/docs/notebooks/segger_tutorial.ipynb
index d5cfc54..6982d34 100644
--- a/docs/notebooks/segger_tutorial.ipynb
+++ b/docs/notebooks/segger_tutorial.ipynb
@@ -213,7 +213,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 8,
+   "execution_count": null,
    "id": "4db89cb4-d0eb-426a-a71f-d127926fa412",
    "metadata": {
     "execution": {
@@ -270,6 +270,11 @@
     "\n",
     "dm.setup()\n",
     "\n",
+    "\n",
+    "# if you wish to use more than 1 device for training you should run this:\n",
+    "batch = dm.train[0]\n",
+    "ls.forward(batch)\n",
+    "\n",
     "# Initialize the Lightning trainer\n",
     "trainer = Trainer(\n",
     "    accelerator='cuda',  \n",