added explicit train and eval mode

Author: lavaman131

lab2/solutions/PT_Part2_Debiasing_Solution.ipynb

@@ -567,6 +567,9 @@
         "if hasattr(tqdm, \"_instances\"):\n",
         "    tqdm._instances.clear()  # clear if it exists\n",
         "\n",
+        "# set the model to train mode\n",
+        "standard_classifier.train()\n",
+        "\n",
         "\n",
         "def standard_train_step(x, y):\n",
         "    x = torch.from_numpy(x).float().to(device)\n",
@@ -635,6 +638,9 @@
       "source": [
         "### Evaluation of standard CNN ###\n",
         "\n",
+        "# set the model to eval mode\n",
+        "standard_classifier.eval()\n",
+        "\n",
         "# TRAINING DATA\n",
         "# Evaluate on a subset of CelebA+Imagenet\n",
         "(batch_x, batch_y) = loader.get_batch(5000)\n",