Merge pull request #37 from ultralytics/master

Set Comet as default logger for Segment, Classify notebooks (ultralytics#11504)
NagatoYuki0943 · May 9, 2023 · 9e6b87c · 9e6b87c
2 parents 6d598f8 + 016e046
commit 9e6b87c
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"
       ]
     },
     {