From d6f93b963ca5603de565123243d13ea6f473f0d9 Mon Sep 17 00:00:00 2001 From: Jason Eshraghian <40262130+jeshraghian@users.noreply.github.com> Date: Sat, 14 Oct 2023 12:55:52 -0700 Subject: [PATCH] Update tutorial_regression_1.ipynb bug-fix: ipynb not opening --- examples/tutorial_regression_1.ipynb | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/examples/tutorial_regression_1.ipynb b/examples/tutorial_regression_1.ipynb index cd19fd7e..960256a3 100644 --- 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" ] }, {