W&B logging bug fix (ultralytics#1399)

assetworx · Nov 17, 2020 · 412b113 · 412b113
1 parent 4250f84
commit 412b113
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test.py b/test.py
@@ -221,7 +221,7 @@ def test(data,
         nt = torch.zeros(1)
 
     # W&B logging
-    if plots and wandb:
+    if plots and wandb and wandb.run:
         wandb.log({"Images": wandb_images})
         wandb.log({"Validation": [wandb.Image(str(x), caption=x.name) for x in sorted(save_dir.glob('test*.jpg'))]})