keep tag_frequency for each dataset

2026-04-14 16:22:28 +00:00 · 2023-03-02 20:27:22 +09:00
parent 83bfb54f20
commit 7e1aa5f4d6
1 changed files with 1 additions and 1 deletions
--- a/train_network.py
+++ b/train_network.py
@@ -316,7 +316,7 @@ def train(args):
        "enable_bucket": bool(dataset.enable_bucket),
        "min_bucket_reso": dataset.min_bucket_reso,
        "max_bucket_reso": dataset.max_bucket_reso,
-        # "tag_frequency": dataset.tag_frequency,
+        "tag_frequency": dataset.tag_frequency,
        "bucket_info": dataset.bucket_info,
      }