fixed error while training model (running acc/loss)

Author: Alvaro Bartolome

notebooks/transfer-learning.ipynb

@@ -715,11 +715,11 @@
     "EPOCHS = 15\n",
     "\n",
     "for epoch in range(1, EPOCHS+1):\n",
-    "    running_loss = .0\n",
-    "    running_corrects = .0\n",
     "    best_acc = .0\n",
     "    print(f\"\\nEpoch {epoch}/{EPOCHS}\\n{'='*25}\")\n",
     "    for phase in ['train', 'val']:\n",
+    "        running_loss = .0\n",
+    "        running_corrects = .0\n",
     "        if phase == 'train': model.train()\n",
     "        if phase == 'val': model.eval()\n",
     "        for inputs, labels in loaders[phase]:\n",
@@ -742,7 +742,7 @@
     "        epoch_acc = running_corrects.double() / dataset_sizes[phase]\n",
     "        if phase == 'train': scheduler.step()\n",
     "        if phase == 'val' and epoch_acc > best_acc:\n",
-    "            best_corrects = epoch_acc\n",
+    "            best_acc = epoch_acc\n",
     "            best_model_weights = deepcopy(model.state_dict())\n",
     "        print(f\"Loss ({phase}): {epoch_loss}, Acc ({phase}): {epoch_acc}\")"
    ]