Fix WandB config improper hierarchical display of keys (ashleve#583)

femto · Jun 21, 2023 · 9355a5d · 9355a5d
1 parent ca07865
commit 9355a5d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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"]