Adding more documentation

examples/mvp.ipynb

@@ -515,7 +515,7 @@
     "        self.ds_test = ds_test\n",
     "\n",
     "    def on_step_end(self, step, loss, compressor, *args, **kargs):\n",
-    "        # Record Loss at every epoch\n",
+    "        # Record Loss at every step\n",
     "        wandb.log({\"train/loss\": loss}, step=step)\n",
     "        if not ((step > 0) and (step % self.log_freq == 0)):\n",
     "            return\n",
@@ -925,24 +925,32 @@
    "id": "6f146264-a34f-4afb-b346-18bd29ef5b29",
    "metadata": {},
    "source": [
-    "## Setup WandB"
+    "## Setup WandB\n",
+    "To login get the api key here: https://wandb.ai/authorize"
    ]
   },
   {
    "cell_type": "code",
    "id": "c8e6f9cc-554c-45a7-bd77-903623e5a402",
    "metadata": {},
+   "source": "wandb.login()",
+   "outputs": [],
+   "execution_count": null
+  },
+  {
+   "metadata": {},
+   "cell_type": "code",
+   "outputs": [],
+   "execution_count": null,
    "source": [
-    "wandb.login()\n",
     "wandb.init(\n",
     "    # set the wandb project where this run will be logged\n",
     "    project=project_name,\n",
     "    # track hyperparameters and run metadata\n",
     "    config=config,\n",
     ")"
    ],
-   "outputs": [],
-   "execution_count": null
+   "id": "444c802afa609c9d"
   },
   {
    "cell_type": "markdown",