Update tutorial_regression_2.ipynb

bug-fix: ipynb file not opening
SpikeAI · Oct 14, 2023 · f2e466e · f2e466e
1 parent d6f93b9
commit f2e466e
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/examples/tutorial_regression_2.ipynb b/examples/tutorial_regression_2.ipynb
@@ -407,8 +407,7 @@
       "outputs": [],
       "source": [
         "hidden = 128\n",
-        "device = torch.device(\"cuda\") if torch.cuda.is_available() else torch.device("mps") if torch.backends.mps.is_available() else torch.device(\"cpu\")\n",
-        "\n",
+        "device = torch.device(\"cuda\") if torch.cuda.is_available() else torch.device(\"cpu\")\n",
         "model = Net(timesteps=num_steps, hidden=hidden, beta=0.9).to(device)"
       ]
     },