Created using Colaboratory

ctjanuhowski · Sep 8, 2022 · 8bfd34b · 8bfd34b
1 parent 1997a71
commit 8bfd34b
Showing 1 changed file with 14 additions and 31 deletions.
diff --git a/tutorial.ipynb b/tutorial.ipynb
@@ -654,46 +654,29 @@
         "<p align=\"\"><a href=\"https://roboflow.com/?ref=ultralytics\"><img width=\"480\" src=\"https://uploads-ssl.webflow.com/5f6bc60e665f54545a1e52a5/6152a275ad4b4ac20cd2e21a_roboflow-annotate.gif\"/></a></p>Label images lightning fast (including with model-assisted labeling)"
       ]
     },
-    {
-      "cell_type": "code",
-      "metadata": {
-        "id": "bOy5KI2ncnWd"
-      },
-      "source": [
-        "# Tensorboard  (optional)\n",
-        "%load_ext tensorboard\n",
-        "%tensorboard --logdir runs/train"
-      ],
-      "execution_count": null,
-      "outputs": []
-    },
     {
       "cell_type": "code",
       "source": [
-        "# ClearML  (optional)\n",
-        "%pip install -q clearml\n",
-        "!clearml-init"
+        "#@title Select YOLOv5 🚀 logger\n",
+        "logger = 'TensorBoard' #@param ['TensorBoard', 'ClearML', 'W&B']\n",
+        "\n",
+        "if logger == 'Tensorboard':\n",
+        "  %load_ext tensorboard\n",
+        "  %tensorboard --logdir runs/train\n",
+        "elif logger == 'ClearML':\n",
+        "  %pip install -q clearml\n",
+        "  !clearml-init\n",
+        "elif logger == 'W&B':\n",
+        "  %pip install -q wandb\n",
+        "  import wandb\n",
+        "  wandb.login()"
       ],
       "metadata": {
-        "id": "DQhI6vvaRWjR"
+        "id": "i3oKtE4g-aNn"
       },
       "execution_count": null,
       "outputs": []
     },
-    {
-      "cell_type": "code",
-      "metadata": {
-        "id": "2fLAV42oNb7M"
-      },
-      "source": [
-        "# Weights & Biases  (optional)\n",
-        "%pip install -q wandb\n",
-        "import wandb\n",
-        "wandb.login()"
-      ],
-      "execution_count": null,
-      "outputs": []
-    },
     {
       "cell_type": "code",
       "metadata": {