Created using Colaboratory

tutorial.ipynb

@@ -563,7 +563,7 @@
         "clear_output()\n",
         "print(f\"Setup complete. Using torch {torch.__version__} ({torch.cuda.get_device_properties(0).name if torch.cuda.is_available() else 'CPU'})\")"
       ],
-      "execution_count": 31,
+      "execution_count": null,
       "outputs": [
         {
           "output_type": "stream",
@@ -889,7 +889,7 @@
         "id": "bOy5KI2ncnWd"
       },
       "source": [
-        "# Tensorboard (optional)\n",
+        "# Tensorboard  (optional)\n",
         "%load_ext tensorboard\n",
         "%tensorboard --logdir runs/train"
       ],
@@ -902,9 +902,10 @@
         "id": "2fLAV42oNb7M"
       },
       "source": [
-        "# Weights & Biases (optional)\n",
-        "%pip install -q wandb  \n",
-        "!wandb login  # use 'wandb disabled' or 'wandb enabled' to disable or enable"
+        "# Weights & Biases  (optional)\n",
+        "%pip install -q wandb\n",
+        "import wandb\n",
+        "wandb.login()"
       ],
       "execution_count": null,
       "outputs": []