experiments done

kmheckel · Feb 28, 2024 · d5caf50 · d5caf50
1 parent c0d43df
commit d5caf50
Show file tree

Hide file tree

Showing 14 changed files with 3,630 additions and 2,046 deletions.
diff --git a/research/paper/NMNIST_jax.ipynb b/research/paper/NMNIST_jax.ipynb
diff --git a/research/paper/NMNIST_jax_refined.ipynb b/research/paper/NMNIST_jax_refined.ipynb
diff --git a/research/paper/NMNIST_jax_test.ipynb b/research/paper/NMNIST_jax_test.ipynb
diff --git a/research/paper/NMNIST_spikingjelly.ipynb b/research/paper/NMNIST_spikingjelly.ipynb
@@ -110,8 +110,7 @@
     "                                      lambda x : np.array(x, dtype=np.uint8)\n",
     "                                     ])\n",
     "\n",
-    "train_dataset = tonic.datasets.NMNIST(save_to='./tmp/data', transform=frame_transform, train=True)\n",
-    "test_dataset = tonic.datasets.NMNIST(save_to='./tmp/data', transform=frame_transform, train=False)"
+    "train_dataset = tonic.datasets.NMNIST(save_to='./tmp/data', transform=frame_transform, train=True)"
    ]
   },
   {
@@ -151,21 +150,6 @@
     "x_train, y_train = x_train.to(device), y_train.to(device)"
    ]
   },
-  {
-   "cell_type": "code",
-   "execution_count": 6,
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "#test_dl = iter(DataLoader(test_dataset, batch_size=len(test_dataset)//10,\n",
-    "#                          collate_fn=tonic.collation.PadTensors(batch_first=True), drop_last=True, shuffle=False))\n",
-    "#        \n",
-    "#x_test, y_test = next(test_dl)\n",
-    "#x_test, y_test = x_test.to(torch.uint8), y_test.to(torch.uint8)\n",
-    "#x_test, y_test = x_test.to(device), y_test.to(device)\n",
-    "#x_test, y_test = shuffle((x_test, y_test))"
-   ]
-  },
   {
    "cell_type": "markdown",
    "metadata": {},

diff --git a/research/paper/NMNIST_torch.ipynb b/research/paper/NMNIST_torch.ipynb
diff --git a/research/paper/SHD_jax.ipynb b/research/paper/SHD_jax.ipynb