Merge pull request #154 from czbiohub-sf/153-issues-when-running-yogo…

…-test 153 issues when running yogo test
czbiohub-sf · Jun 17, 2024 · be9dc41 · Axel-Jacobsen · Jun 19, 2024 · be9dc41
2 parents cef8673 + a5cda91
commit be9dc41
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/yogo/model.py b/yogo/model.py
@@ -143,6 +143,7 @@ def from_pth(
         return model, {
             "step": global_step,
             "class_names": class_names,
+            "normalize_images": params["normalize_images"],
         }
 
     def to(self, device, *args, **kwargs):

diff --git a/yogo/utils/test_model.py b/yogo/utils/test_model.py
@@ -40,7 +40,7 @@ def test_model(args: argparse.Namespace) -> None:
         "slurm-job-id": os.getenv("SLURM_JOB_ID", default=None),
     }
 
-    log_to_wandb = args.wandb or len(args.wandb_resume_id) > 0
+    log_to_wandb = args.wandb or (args.wandb_resume_id is not None)
 
     if log_to_wandb:
         print("logging to wandb")