W&B: DDP fix (ultralytics#5176)

* fix dpp with wandb disabled * PyCharm reformat * Fix * fix Co-authored-by: Glenn Jocher <[email protected]>
ZongqingHou · Oct 14, 2021 · 1172ac1 · 1172ac1
1 parent e520e82
commit 1172ac1
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://'