Fix bug. Fix batch_size bug.

Author: liuwake

transfer-learning/Transfer_Learning_Exercise.ipynb

@@ -382,7 +382,7 @@
     "    correct_tensor = pred.eq(target.data.view_as(pred))\n",
     "    correct = np.squeeze(correct_tensor.numpy()) if not train_on_gpu else np.squeeze(correct_tensor.cpu().numpy())\n",
     "    # calculate test accuracy for each object class\n",
-    "    for i in range(batch_size):\n",
+    "    for i in range(len(target)):\n",
     "        label = target.data[i]\n",
     "        class_correct[label] += correct[i].item()\n",
     "        class_total[label] += 1\n",

transfer-learning/Transfer_Learning_Solution.ipynb

@@ -532,7 +532,7 @@
     "    correct_tensor = pred.eq(target.data.view_as(pred))\n",
     "    correct = np.squeeze(correct_tensor.numpy()) if not train_on_gpu else np.squeeze(correct_tensor.cpu().numpy())\n",
     "    # calculate test accuracy for each object class\n",
-    "    for i in range(batch_size):\n",
+    "    for i in range(len(target)):\n",
     "        label = target.data[i]\n",
     "        class_correct[label] += correct[i].item()\n",
     "        class_total[label] += 1\n",