updated link to data folder

Author: leriomaggio

1-fast-gradient-sign-method/FSGM-Attack.ipynb

@@ -3201,7 +3201,16 @@
     "from pathlib import Path\n",
     "import os \n",
     "\n",
-    "DATA_FOLDER = Path(os.path.join(os.path.abspath(os.path.curdir), \"..\", \"..\")) / \"data\""
+    "DATA_FOLDER = Path(os.path.join(os.path.abspath(os.path.curdir), \"..\")) / \"data\""
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "print(DATA_FOLDER)"
    ]
   },
   {
@@ -3268,9 +3277,22 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "PRETRAINED_MODEL_WEIGHTS = \"lenet_mnist_model.pth\"\n",
+    "if torch.cuda.is_available():\n",
+    "    dev_name = \"cuda\"\n",
+    "else:\n",
+    "    dev_name = \"cpu\"\n",
     "\n",
-    "device = torch.device(\"cuda\" if torch.cuda.is_available() else \"cpu\")\n",
+    "device = torch.device(dev_name)\n",
+    "print(f\"You will be using the {device} device\")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "PRETRAINED_MODEL_WEIGHTS = \"lenet_mnist_model.pth\"\n",
     "\n",
     "# Initialize the network\n",
     "model = Net().to(device)\n",
@@ -3519,7 +3541,9 @@
     "\n",
     "# Run test for each epsilon\n",
     "for ε in EPSILONS[1:]:\n",
-    "    # your code here"
+    "    acc, adv_examples = test(model, device, test_loader, ε)\n",
+    "    accuracies.append(acc)\n",
+    "    adv_examples_map[ε] = adv_examples"
    ]
   },
   {
@@ -3605,7 +3629,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.10.5"
+   "version": "3.9.16"
   }
  },
  "nbformat": 4,