changing loss visualization

Author: slolla

lab2/solutions/Part2_Debiasing_Solution.ipynb

@@ -1344,11 +1344,15 @@
         "for i in range(1, 6):\n",
         "  axs[i - 1].imshow(test_faces[- 1 * i])\n",
         "  axs[i - 1].set_title(round(vae_loss.numpy()[-1 * i], 7))\n",
+        "print(\"Samples with the highest reconstruction loss\")\n",
+        "plt.show()\n",
         "\n",
         "fig, axs = plt.subplots(1, 5)\n",
         "for i in range(5):\n",
         "  axs[i].imshow(test_faces[i])\n",
-        "  axs[i].set_title(round(vae_loss.numpy()[i], 7))"
+        "  axs[i].set_title(round(vae_loss.numpy()[i], 7))\n",
+        "print(\"Samples with the lowest reconstruction loss\")\n",
+        "plt.show()"
       ]
     },
     {