Fix band weights via toml. Add more logging

2026-04-17 17:24:21 +00:00 · 2025-04-11 20:39:31 -04:00
parent 20a99771bf
commit 7b9e92a8cc
2 changed files with 8 additions and 2 deletions
--- a/library/train_util.py
+++ b/library/train_util.py
@@ -4657,6 +4657,10 @@ def read_config_from_file(args: argparse.Namespace, parser: argparse.ArgumentPar
            ignore_nesting_dict[section_name] = section_dict
            continue

+        if section_name == "wavelet_loss_band_weights":
+            ignore_nesting_dict[section_name] = section_dict
+            continue
+
        # if value is dict, save all key and value into one dict
        for key, value in section_dict.items():
            ignore_nesting_dict[key] = value