Add scale map to max_norm

2026-04-17 17:24:21 +00:00 · 2025-01-11 13:42:20 -05:00
parent e89653975d
commit b0d0d43bfa
3 changed files with 28 additions and 2 deletions
--- a/library/train_util.py
+++ b/library/train_util.py
@@ -4537,6 +4537,10 @@ def read_config_from_file(args: argparse.Namespace, parser: argparse.ArgumentPar
            ignore_nesting_dict[section_name] = section_dict
            continue

+        if section_name == "scale_weight_norms_map":
+            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