fixed ignore_index bug

VMarsocci · Sep 11, 2024 · 8ea3149 · 8ea3149
1 parent ff8c83b
commit 8ea3149
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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
+    return dict(nested) if nested else None