permit to log visualization result to disk

microsoft · Nov 29, 2021 · 4a0553f · 4a0553f
1 parent a872c20
commit 4a0553f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ssod/utils/logger.py b/ssod/utils/logger.py
@@ -109,7 +109,7 @@ def log_image_with_boxes(
     if not (interval == 1 or _log_counter[key] % interval == 1):
         return
     if backend == "auto":
-        if wandb is None:
+        if (wandb is None) or (wandb.run is None):
             backend = "file"
         else:
             backend = "wandb"