From f441c90503ad88b02dd53f7e0296f9f519bec23a Mon Sep 17 00:00:00 2001 From: Johan MEJIA Date: Tue, 28 Sep 2021 22:21:42 +0200 Subject: [PATCH] fix bug in load_training --- alonet/common/pl_helpers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/alonet/common/pl_helpers.py b/alonet/common/pl_helpers.py index e5fbb203..0b0ab349 100644 --- a/alonet/common/pl_helpers.py +++ b/alonet/common/pl_helpers.py @@ -93,7 +93,7 @@ def load_training( if "weights" in kwargs and kwargs["weights"] is not None: # Highest priority weights_path = kwargs["weights"] - strict = True if "nostrict" in args else not args.nostrict + strict = True if "nostrict" not in args else not args.nostrict if run_id is not None and project_run_id is not None: run_id_project_dir = os.path.join(vb_folder(), f"project_{project_run_id}") ckpt_path = os.path.join(run_id_project_dir, run_id, "last.ckpt")