Update message

utils/loggers/__init__.py

@@ -5,9 +5,9 @@
 import torch
 from torch.utils.tensorboard import SummaryWriter
 
-from utils.general import colorstr
-from utils.torch_utils import de_parallel
+from utils.general import colorstr, emojis
 from utils.loggers.wandb.wandb_utils import WandbLogger
+from utils.torch_utils import de_parallel
 
 LOGGERS = ('txt', 'tb', 'wandb')  # text-file, TensorBoard, Weights & Biases
 
@@ -37,6 +37,14 @@ def __init__(self, save_dir=None, results_file=None, weights=None, opt=None, hyp
     def start(self):
         self.txt = True  # always log to txt
 
+        # Message
+        try:
+            import wandb
+        except ImportError:
+            prefix = colorstr('Weights & Biases: ')
+            s = f"{prefix}run 'pip install wandb' to automatically track and visualize YOLOv5 🚀 runs (RECOMMENDED)"
+            print(emojis(s))
+
         # TensorBoard
         s = self.save_dir
         if 'tb' in self.include and not self.opt.evolve:

utils/loggers/wandb/wandb_utils.py

@@ -1,4 +1,5 @@
 """Utilities and tools for tracking runs with Weights & Biases."""
+
 import logging
 import os
 import sys
@@ -13,16 +14,13 @@
 
 from utils.datasets import LoadImagesAndLabels
 from utils.datasets import img2label_paths
-from utils.general import colorstr, check_dataset, check_file, emojis
+from utils.general import check_dataset, check_file
 
 try:
     import wandb
 
     assert hasattr(wandb, '__version__')  # verify package import not local dir
 except (ImportError, AssertionError):
-    prefix = colorstr('Weights & Biases: ')
-    s = f"{prefix}run 'pip install wandb' to automatically track and visualize YOLOv5 🚀 training (RECOMMENDED)"
-    print(emojis(s))
     wandb = None
 
 RANK = int(os.getenv('RANK', -1))