Set Comet as default logger for Segment, Classify notebooks (#11504)

* Created using Colaboratory * Created using Colaboratory * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
ultralytics · May 8, 2023 · 016e046 · 016e046
1 parent c3e4e94
commit 016e046
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 12 deletions.
diff --git a/classify/tutorial.ipynb b/classify/tutorial.ipynb
@@ -1255,16 +1255,17 @@
       "outputs": [],
       "source": [
         "#@title Select YOLOv5 🚀 logger {run: 'auto'}\n",
-        "logger = 'TensorBoard' #@param ['TensorBoard', 'Comet', 'ClearML']\n",
+        "logger = 'Comet' #@param ['Comet', 'ClearML', 'TensorBoard']\n",
         "\n",
-        "if logger == 'TensorBoard':\n",
-        "  %load_ext tensorboard\n",
-        "  %tensorboard --logdir runs/train\n",
-        "elif logger == 'Comet':\n",
+        "if logger == 'Comet':\n",
         "  %pip install -q comet_ml\n",
         "  import comet_ml; comet_ml.init()\n",
         "elif logger == 'ClearML':\n",
-        "  import clearml; clearml.browser_login()"
+        "  %pip install -q clearml\n",
+        "  import clearml; clearml.browser_login()\n",
+        "elif logger == 'TensorBoard':\n",
+        "  %load_ext tensorboard\n",
+        "  %tensorboard --logdir runs/train"
       ]
     },
     {

diff --git a/segment/tutorial.ipynb b/segment/tutorial.ipynb
@@ -250,16 +250,17 @@
       "outputs": [],
       "source": [
         "#@title Select YOLOv5 🚀 logger {run: 'auto'}\n",
-        "logger = 'TensorBoard' #@param ['TensorBoard', 'Comet', 'ClearML']\n",
+        "logger = 'Comet' #@param ['Comet', 'ClearML', 'TensorBoard']\n",
         "\n",
-        "if logger == 'TensorBoard':\n",
-        "  %load_ext tensorboard\n",
-        "  %tensorboard --logdir runs/train-seg\n",
-        "elif logger == 'Comet':\n",
+        "if logger == 'Comet':\n",
         "  %pip install -q comet_ml\n",
         "  import comet_ml; comet_ml.init()\n",
         "elif logger == 'ClearML':\n",
-        "  import clearml; clearml.browser_login()"
+        "  %pip install -q clearml\n",
+        "  import clearml; clearml.browser_login()\n",
+        "elif logger == 'TensorBoard':\n",
+        "  %load_ext tensorboard\n",
+        "  %tensorboard --logdir runs/train"
       ]
     },
     {