Add tag frequency metadata

2026-04-08 22:35:09 +00:00 · 2023-01-23 17:43:03 -08:00
parent 93df55d597
commit 2e8a3d20dd
2 changed files with 11 additions and 0 deletions
--- a/train_network.py
+++ b/train_network.py
@@ -264,6 +264,7 @@ def train(args):
      "ss_keep_tokens": args.keep_tokens,
      "ss_dataset_dirs": json.dumps(train_dataset.dataset_dirs_info),
      "ss_reg_dataset_dirs": json.dumps(train_dataset.reg_dataset_dirs_info),
+      "ss_tag_frequency": json.dumps(train_dataset.tag_frequency),
      "ss_training_comment": args.training_comment        # will not be updated after training
  }