4 is batchSize at line 901

maalvarezl · Dec 9, 2020 · 8017fcc · 8017fcc
1 parent 583c111
commit 8017fcc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Lab 7 - Neural Networks.ipynb b/Lab 7 - Neural Networks.ipynb
@@ -898,7 +898,7 @@
     "        outputs = loadCNN(images)\n",
     "        _, predicted = torch.max(outputs, 1)\n",
     "        c = (predicted == labels).squeeze()\n",
-    "        for i in range(4):\n",
+    "        for i in range(batchSize):\n",
     "            label = labels[i]\n",
     "            class_correct[label] += c[i].item()\n",
     "            class_total[label] += 1\n",