Update tutorial_sae.ipynb

bug-fix: ipynb file rollback
SpikeAI · Oct 14, 2023 · 270515f · 270515f
1 parent f2e466e
commit 270515f
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/examples/tutorial_sae.ipynb b/examples/tutorial_sae.ipynb
@@ -199,10 +199,10 @@
    "source": [
     "# dataloader arguments\n",
     "batch_size = 250\n",
-    "data_path='/tmp/data/mnist'\n",
+    "data_path='/data/mnist'\n",
     "\n",
     "dtype = torch.float\n",
-    "device = torch.device(\"cuda\") if torch.cuda.is_available() else torch.device("mps") if torch.backends.mps.is_available() else torch.device(\"cpu\")"
+    "device = torch.device(\"cuda\") if torch.cuda.is_available() else torch.device(\"cpu\")\n"
    ]
   },
   {
@@ -762,7 +762,7 @@
     "\n",
     "#setup GPU\n",
     "dtype = torch.float\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",
+    "device = torch.device(\"cuda\") if torch.cuda.is_available() else torch.device(\"cpu\")\n",
     "\n",
     "# neuron and simulation parameters\n",
     "spike_grad = surrogate.atan(alpha=2.0)# alternate surrogate gradient fast_sigmoid(slope=25) \n",