diff --git a/examples/dataset_examples/SNN_Tutorial_Derrick_Lee.ipynb b/examples/dataset_examples/SNN_Tutorial_Derrick_Lee.ipynb
index 089b0e7f..751db823 100644
--- a/examples/dataset_examples/SNN_Tutorial_Derrick_Lee.ipynb
+++ b/examples/dataset_examples/SNN_Tutorial_Derrick_Lee.ipynb
@@ -7,7 +7,7 @@
"colab_type": "text"
},
"source": [
- ""
+ ""
]
},
{
@@ -57,23 +57,23 @@
},
{
"cell_type": "code",
- "execution_count": null,
+ "execution_count": 1,
"metadata": {
"id": "KJAF2tZPvPMk",
"colab": {
"base_uri": "https://localhost:8080/"
},
- "outputId": "8e60c8cd-1f7e-4c56-f60a-e4fd0da674bf"
+ "outputId": "d6e97bb2-0133-4dcd-c5bb-0f6801dd3b80"
},
"outputs": [
{
"output_type": "stream",
"name": "stdout",
"text": [
- "\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m109.7/109.7 kB\u001b[0m \u001b[31m2.6 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
- "\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m112.7/112.7 kB\u001b[0m \u001b[31m9.8 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
- "\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m50.4/50.4 kB\u001b[0m \u001b[31m6.0 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
- "\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m109.0/109.0 kB\u001b[0m \u001b[31m3.0 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
+ "\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m109.7/109.7 kB\u001b[0m \u001b[31m3.3 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
+ "\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m107.5/107.5 kB\u001b[0m \u001b[31m7.7 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
+ "\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m50.4/50.4 kB\u001b[0m \u001b[31m6.5 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
+ "\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m109.0/109.0 kB\u001b[0m \u001b[31m1.5 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
"\u001b[?25h"
]
}
@@ -98,12 +98,14 @@
"from snntorch import surrogate\n",
"from snntorch import functional as SF\n",
"from snntorch import utils\n",
- "import torch.nn as nn"
+ "import torch.nn as nn\n",
+ "\n",
+ "device = torch.device(\"cuda\") if torch.cuda.is_available() else torch.device(\"mps\") if torch.backends.mps.is_available() else torch.device(\"cpu\")"
],
"metadata": {
"id": "C9F9ihlGls9u"
},
- "execution_count": null,
+ "execution_count": 2,
"outputs": []
},
{
@@ -141,30 +143,30 @@
"metadata": {
"colab": {
"base_uri": "https://localhost:8080/",
- "height": 156,
+ "height": 86,
"referenced_widgets": [
- "3974416247f7451bb2bb82bdb83138fc",
- "0f504b5f4ad74595ae37ebf795d46514",
- "5fd4888429c342f2b1493c1689e6fc6b",
- "b4a14c5ad41441a7bbb76aa96f72778b",
- "766a82612ccc442fb46264421109c713",
- "92e3de6e58a24ea1a870432e8d907504",
- "c0c3f6d088534ceb9d70af69d7e08392",
- "0b28db65ac3743c2abb7ddd085fbed21",
- "1071d5988c9b4cf7b846b51db4f66a32",
- "daa45045ae794990867fce4f0be28245",
- "c710df2cd11e4ec2b1c9da6585c76016"
+ "ab87bfd9c02540f58c277f34bf624b84",
+ "449eecdf6da24230adc13f0316e005a4",
+ "c1964cca2dd34ee6adeebc592d3d0e8f",
+ "a310e770aace454686dc70c1e6f04474",
+ "61016fb4e16346f7a12a03f2a5a5c669",
+ "5f23795f6bc44987bb3f81ac428b0356",
+ "8041b1851a1e462d94c0eb24963d4a6a",
+ "322859851f804d67a53b357e63a8227e",
+ "44fa051f61a14aef894a656887ea3f28",
+ "8d0fa726820b47698b80e8eb31b7be41",
+ "c9285a1279ce44b3ba3ad73f70c6c86f"
]
},
"id": "peDXjxJUwEVX",
- "outputId": "0a85d931-90d1-4fc4-8867-32266e8cd4fb"
+ "outputId": "d65336a9-37a1-4f15-f9d8-2f404d8fa21e"
},
"outputs": [
{
"output_type": "stream",
"name": "stdout",
"text": [
- "Downloading https://s3-eu-west-1.amazonaws.com/pfigshare-u-files/38022171/ibmGestureTrain.tar.gz?X-Amz-Algorithm=AWS4-HMAC-SHA256&X-Amz-Credential=AKIAIYCQYOYV5JSSROOA/20231102/eu-west-1/s3/aws4_request&X-Amz-Date=20231102T212615Z&X-Amz-Expires=10&X-Amz-SignedHeaders=host&X-Amz-Signature=587183aaf756ad2dc11fa835233d8e4b110563001b4e86117a308156d072b349 to ./data/DVSGesture/ibmGestureTrain.tar.gz\n"
+ "Downloading https://s3-eu-west-1.amazonaws.com/pfigshare-u-files/38022171/ibmGestureTrain.tar.gz?X-Amz-Algorithm=AWS4-HMAC-SHA256&X-Amz-Credential=AKIAIYCQYOYV5JSSROOA/20231107/eu-west-1/s3/aws4_request&X-Amz-Date=20231107T212148Z&X-Amz-Expires=10&X-Amz-SignedHeaders=host&X-Amz-Signature=ceb2bd6df9661156db57c6d97af632ba756bd128bab1757a48835be033fb3dcc to ./data/DVSGesture/ibmGestureTrain.tar.gz\n"
]
},
{
@@ -176,20 +178,10 @@
"application/vnd.jupyter.widget-view+json": {
"version_major": 2,
"version_minor": 0,
- "model_id": "3974416247f7451bb2bb82bdb83138fc"
+ "model_id": "ab87bfd9c02540f58c277f34bf624b84"
}
},
"metadata": {}
- },
- {
- "output_type": "stream",
- "name": "stdout",
- "text": [
- "Extracting ./data/DVSGesture/ibmGestureTrain.tar.gz to ./data/DVSGesture\n",
- "Dataset contains 1077 samples.\n",
- "There are 787770 events in the first sample.\n",
- "A single event: (119, 113, False, 6)\n"
- ]
}
],
"source": [
@@ -219,41 +211,9 @@
"cell_type": "code",
"execution_count": null,
"metadata": {
- "colab": {
- "base_uri": "https://localhost:8080/"
- },
- "id": "tIUgiH4z52IC",
- "outputId": "386e6579-be2f-46ba-d5a6-3424cad3878e"
+ "id": "tIUgiH4z52IC"
},
- "outputs": [
- {
- "output_type": "stream",
- "name": "stdout",
- "text": [
- "(787, 2, 128, 128)\n"
- ]
- },
- {
- "output_type": "execute_result",
- "data": {
- "text/plain": [
- ""
- ]
- },
- "metadata": {},
- "execution_count": 4
- },
- {
- "output_type": "display_data",
- "data": {
- "text/plain": [
- "