From 9355a5de241dfa2eea721c94e73747b01f5bbff5 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Andr=C3=A9=20Aquilina?=
 <32460579+dreaquil@users.noreply.github.com>
Date: Wed, 21 Jun 2023 17:37:40 +0200
Subject: [PATCH] Fix WandB config improper hierarchical display of keys (#583)

---
 src/utils/logging_utils.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

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"]