From 8ea3149a47e58895e8e5323d5ac03ece56e93eb1 Mon Sep 17 00:00:00 2001
From: Valerio Marsocci <49810041+VMarsocci@users.noreply.github.com>
Date: Wed, 11 Sep 2024 15:24:47 +0200
Subject: [PATCH] fixed ignore_index bug

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

diff --git a/utils/configs.py b/utils/configs.py
index a581c301..2b0f8ebf 100644
--- a/utils/configs.py
+++ b/utils/configs.py
@@ -39,6 +39,7 @@ def load_configs(parser:argparse.ArgumentParser) -> OmegaConf:
             encoder_cfg["input_bands"] = dataset_cfg['bands']
 
         segmentor_cfg["loss"]['distribution'] = dataset_cfg['distribution']
+        segmentor_cfg["loss"]['ignore_index'] = dataset_cfg['ignore_index']
 
         file_cfg["encoder"] = encoder_cfg
         file_cfg["dataset"] = dataset_cfg
@@ -119,4 +120,4 @@ def _nest(
             if val is None and include_none:
                 nested[key] = val
 
-    return dict(nested) if nested else None
\ No newline at end of file
+    return dict(nested) if nested else None