modified inference notebook

IvanLauLinTiong · Mar 18, 2022 · 2d2e0ef · 2d2e0ef
1 parent d9bb414
commit 2d2e0ef
Showing 1 changed file with 11 additions and 5 deletions.
diff --git a/inference.ipynb b/inference.ipynb
@@ -72,7 +72,8 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "MODEL_PATH = \"./model/model_epoch50_lr0.01_batch8.pt\""
+    "# MODEL_PATH = \"./model/model_epoch50_lr0.01_batch8.pt\"\n",
+    "MODEL_PATH = \"./model_de_Mar19_01-27-40_epoch25_lr0.005_batch128_acc0.481.pt\""
    ]
   },
   {
@@ -141,12 +142,17 @@
    "outputs": [],
    "source": [
     "# tranformations\n",
+    "# preprocess = transforms.Compose([\n",
+    "#     transforms.ToPILImage(),\n",
+    "#     transforms.Grayscale(3),\n",
+    "#     transforms.Resize(256),\n",
+    "#     transforms.ToTensor(),\n",
+    "#     transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])\n",
+    "# ])\n",
+    "\n",
     "preprocess = transforms.Compose([\n",
-    "    transforms.ToPILImage(),\n",
-    "    transforms.Grayscale(3),\n",
-    "    transforms.Resize(256),\n",
     "    transforms.ToTensor(),\n",
-    "    transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])\n",
+    "    transforms.Normalize((0.5,), (0.5,))\n",
     "])"
    ]
   },