From 7e1aa5f4d640f4781d4aecf2aca798752ca45426 Mon Sep 17 00:00:00 2001
From: Kohya S <ykumeykume@gmail.com>
Date: Thu, 2 Mar 2023 20:27:22 +0900
Subject: [PATCH] keep tag_frequency for each dataset

---
 train_network.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/train_network.py b/train_network.py
index 9dac2ed3..5cd08d49 100644
--- 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,
       }