From 412b113ac694d8fca9253668dad78f3550d81ea8 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Tue, 17 Nov 2020 19:17:57 +0100 Subject: [PATCH] W&B logging bug fix (#1399) --- test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test.py b/test.py index f627475..b4c97bf 100644 --- 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'))]})