Improve docs

theislab · Nov 21, 2024 · 39f0add · 39f0add
1 parent a79ebfc
commit 39f0add
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/docs/notebooks/general_pipeline.ipynb b/docs/notebooks/general_pipeline.ipynb
@@ -489,10 +489,11 @@
     "# For cpu training you should add the following line to the model.train parameters:\n",
     "# accelerator=\"cpu\", devices=1,\n",
     "#\n",
-    "# For cpu training you should add the following line to the model.train parameters:\n",
+    "# For mps acceleration on macbooks, add the following line to the model.train parameters:\n",
     "# accelerator=\"mps\", devices=1,\n",
     "#\n",
     "# For gpu training don't provide any additional parameter.\n",
+    "# More details here: https://lightning.ai/docs/pytorch/stable/accelerators/gpu_basic.html\n",
     "\n",
     "n_epochs = 400\n",
     "\n",

diff --git a/docs/notebooks/general_pipeline.py b/docs/notebooks/general_pipeline.py
@@ -132,10 +132,11 @@
 # For cpu training you should add the following line to the model.train parameters:
 # accelerator="cpu", devices=1,
 #
-# For cpu training you should add the following line to the model.train parameters:
+# For mps acceleration on macbooks, add the following line to the model.train parameters:
 # accelerator="mps", devices=1,
 #
 # For gpu training don't provide any additional parameter.
+# More details here: https://lightning.ai/docs/pytorch/stable/accelerators/gpu_basic.html
 
 n_epochs = 400