diff --git a/src/utils/logging_utils.py b/src/utils/logging_utils.py index 276323495..3a8ca8e5c 100644 --- a/src/utils/logging_utils.py +++ b/src/utils/logging_utils.py @@ -1,4 +1,5 @@ from lightning.pytorch.utilities import rank_zero_only +from omegaconf import OmegaConf from src.utils import pylogger @@ -15,7 +16,7 @@ def log_hyperparameters(object_dict: dict) -> None: hparams = {} - cfg = object_dict["cfg"] + cfg = OmegaConf.to_container(object_dict["cfg"]) model = object_dict["model"] trainer = object_dict["trainer"]