quickstart rollback

bug-fix: ipynb file not opening
jeshraghian · Oct 14, 2023 · 950878d · 950878d
1 parent ed48a10
commit 950878d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/examples/quickstart.ipynb b/examples/quickstart.ipynb
@@ -88,8 +88,8 @@
       "outputs": [],
       "source": [
         "batch_size = 128\n",
-        "data_path='/tmp/data/mnist'\n",
-        "device = torch.device(\"cuda\") if torch.cuda.is_available() else torch.device("mps") if torch.backends.mps.is_available() else torch.device(\"cpu\")"
+        "data_path='/data/mnist'\n",
+        "device = torch.device(\"cuda\") if torch.cuda.is_available() else torch.device(\"cpu\")"
       ]
     },
     {