W&B logging bug fix (#1399)

2020-11-17 19:17:57 +01:00 · 2020-11-17 19:17:57 +01:00 · 412b113ac6
parent 4250f84dfb
commit 412b113ac6
1 changed files with 1 additions and 1 deletions
--- 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'))]})