Update tutorial_regression_1.ipynb

bug-fix: ipynb not opening
jeshraghian · Oct 14, 2023 · d6f93b9 · d6f93b9
1 parent 950878d
commit d6f93b9
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/examples/tutorial_regression_1.ipynb b/examples/tutorial_regression_1.ipynb
@@ -387,9 +387,8 @@
    "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",
-    "model = Net(timesteps=num_steps, hidden=hidden).to(device)"
+    "device = torch.device(\"cuda\") if torch.cuda.is_available() else torch.device(\"cpu\")\n",
+    "model = Net(timesteps=num_steps, hidden=hidden).to(device)\n"
    ]
   },
   {