Add bucketing metadata

2026-04-08 22:35:09 +00:00 · 2023-01-23 17:26:58 -08:00
parent 93df55d597
commit 66051883fb
2 changed files with 5 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_bucket_info": json.dumps(train_dataset.bucket_info),
      "ss_training_comment": args.training_comment        # will not be updated after training
  }