From 48fb92d6fd65e6b5da7d2bdaea851cf03ffdbc2d Mon Sep 17 00:00:00 2001 From: Fatwir Mohammed Date: Tue, 4 Feb 2025 15:22:04 -0800 Subject: [PATCH] minor bugfix for the ckpt config --- cyto_dl/utils/template_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cyto_dl/utils/template_utils.py b/cyto_dl/utils/template_utils.py index 78bd0b85..1bf57c2f 100644 --- a/cyto_dl/utils/template_utils.py +++ b/cyto_dl/utils/template_utils.py @@ -181,7 +181,7 @@ def log_hyperparameters(object_dict: dict) -> None: hparams["task_name"] = cfg.get("task_name") hparams["tags"] = cfg.get("tags") - hparams["ckpt_path"] = cfg.checkpoint.get("ckpt_path") + hparams["ckpt_path"] = cfg.get('checkpoint', {}).get('ckpt_path') hparams["seed"] = cfg.get("seed") try: