comet updates, sending plot to comet

Author: avaamini

lab1/solutions/Part1_TensorFlow_Solution.ipynb

@@ -86,7 +86,9 @@
         "When training models, it can be useful to visualize information about the model with plots. We can do this manually, but here, we'll show you how to do this using a tool called [Comet ML](https://www.comet.com/docs/v2/), which generates loss and GPU usage curves for you. As you'll see, Comet also enables easy saving of your experiments to a central interface.\n",
         "\n",
         "First, sign up for a Comet account [at this link](https://www.comet.com/signup?utm_source=mit_dl&utm_medium=partner&utm_content=github\n",
-        ") (you can use your Google or Github account). Running this cell will prompt you to enter your API Key, which you can find either in the first 'Get Started with Comet' page or by pressing the '?' in the top right corner and then 'Quickstart Guide' -- it is on the right hand side of the page."
+        ") (you can use your Google or Github account). Running this cell will prompt you to enter your API Key, which you can find either in the first 'Get Started with Comet' page or by pressing the '?' in the top right corner and then 'Quickstart Guide' -- it is on the right hand side of the page.\n",
+        "\n",
+        "After entering your API key, you will find a unique link taking you to a project page for this project -- which we have named `6.s191_lab1_part1`. Note that you can find your active projects under your home `projects` tab in the Comet web interface."
       ]
     },
     {
@@ -705,7 +707,8 @@
         "plt.legend(('Predicted', 'True'))\n",
         "plt.xlabel('Iteration')\n",
         "plt.ylabel('x value')\n",
-        "comet_experiment.end()"
+        "# log the plot to comet\n",
+        "comet_experiment.log_figure(figure_name=\"optimizing\", figure=plt)"
       ]
     },
     {
@@ -718,6 +721,18 @@
         "\n",
         "In the above code block, you also saw how you can directly log metrics, like loss values, to Comet. These are then retained within your profile's web interface, where you are able to visualize the results of your different training runs!"
       ]
+    },
+    {
+      "cell_type": "code",
+      "source": [
+        "# now that we are done, we can end the comet experiment!\n",
+        "comet_experiment.end()"
+      ],
+      "metadata": {
+        "id": "Ba2YF7EN0QPz"
+      },
+      "execution_count": null,
+      "outputs": []
     }
   ],
   "metadata": {