added lines to the tutorial

EliHei2 · Sep 24, 2024 · bf9ce8e · bf9ce8e
1 parent 0184f3d
commit bf9ce8e
Showing 1 changed file with 6 additions and 1 deletion.
diff --git 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",