ODIN

dataflowr · Feb 28, 2024 · 0c141c3 · 0c141c3
1 parent bb2583a
commit 0c141c3
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/Module17/ODIN_mobilenet_empty.ipynb b/Module17/ODIN_mobilenet_empty.ipynb
@@ -108,7 +108,9 @@
     "NORM_SCALE = [63.0 / 255, 62.1 / 255.0, 66.7 / 255.0]\n",
     "EPS_FSGM = 1e-2\n",
     "IMAGE_SIZE = (32, 32)\n",
-    "batch_size = 64"
+    "batch_size = 64\n",
+    "\n",
+    "%mkdir data"
    ]
   },
   {
@@ -126,7 +128,7 @@
     "    ]\n",
     ")\n",
     "\n",
-    "testset = torchvision.datasets.CIFAR10(root=\"/home/mlelarge/data/\", train=False, download=True, transform=transform)\n",
+    "testset = torchvision.datasets.CIFAR10(root=\"./data/\", train=False, download=True, transform=transform)\n",
     "testloaderIn = DataLoader(testset, batch_size=batch_size, shuffle=False, num_workers=2)"
    ]
   },
@@ -233,7 +235,6 @@
    "source": [
     "# The code below dowload the out-of-distribution dataset\n",
     "# if on colab first uncomment and run the following command (only once):\n",
-    "#%mkdir data\n",
     "#%cd data\n",
     "#!wget https://www.dropbox.com/s/avgm2u562itwpkl/Imagenet.tar.gz\n",
     "#!tar -xvzf Imagenet.tar.gz\n",