tutorial 6 rollback

bug fix that prevented ipynb file from opening
jeshraghian · Oct 14, 2023 · 3c85b52 · 3c85b52
1 parent 7d838d9
commit 3c85b52
Showing 1 changed file with 12 additions and 12 deletions.
diff --git a/examples/tutorial_6_CNN.ipynb b/examples/tutorial_6_CNN.ipynb
@@ -3,8 +3,8 @@
     {
       "cell_type": "markdown",
       "metadata": {
-        "colab_type": "text",
-        "id": "view-in-github"
+        "id": "view-in-github",
+        "colab_type": "text"
       },
       "source": [
         "<a href=\"https://colab.research.google.com/github/jeshraghian/snntorch/blob/docs-update/examples/tutorial_6_CNN.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
@@ -261,10 +261,10 @@
       "source": [
         "# dataloader arguments\n",
         "batch_size = 128\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\")"
       ]
     },
     {
@@ -630,11 +630,6 @@
     },
     {
       "cell_type": "code",
-      "execution_count": null,
-      "metadata": {
-        "id": "y_VQ9es-gSO3"
-      },
-      "outputs": [],
       "source": [
         "optimizer = torch.optim.Adam(net.parameters(), lr=1e-2, betas=(0.9, 0.999))\n",
         "num_epochs = 1\n",
@@ -676,7 +671,12 @@
         "              test_acc_hist.append(test_acc.item())\n",
         "\n",
         "        counter += 1"
-      ]
+      ],
+      "metadata": {
+        "id": "y_VQ9es-gSO3"
+      },
+      "execution_count": null,
+      "outputs": []
     },
     {
       "cell_type": "markdown",
@@ -816,9 +816,9 @@
   "metadata": {
     "accelerator": "GPU",
     "colab": {
-      "include_colab_link": true,
       "name": "tutorial_6_CNN.ipynb",
-      "provenance": []
+      "provenance": [],
+      "include_colab_link": true
     },
     "kernelspec": {
       "display_name": "Python 3",