W&B: DDP fix (ultralytics#5176)

* fix dpp with wandb disabled * PyCharm reformat * Fix * fix Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
fhkiel-mlaip · Oct 14, 2021 · 5563986 · 5563986
1 parent c70113b
commit 5563986
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/utils/loggers/wandb/wandb_utils.py b/utils/loggers/wandb/wandb_utils.py
@@ -19,6 +19,13 @@
 from utils.datasets import img2label_paths
 from utils.general import check_dataset, check_file
 
+try:
+    import wandb
+
+    assert hasattr(wandb, '__version__')  # verify package import not local dir
+except (ImportError, AssertionError):
+    wandb = None
+
 RANK = int(os.getenv('RANK', -1))
 WANDB_ARTIFACT_PREFIX = 'wandb-artifact://'