From b03721b4d971cc15ea8f40c274223425fe0d9804 Mon Sep 17 00:00:00 2001 From: Kohya S Date: Fri, 10 Feb 2023 17:36:38 +0900 Subject: [PATCH] Add todo comment --- library/train_util.py | 1 + 1 file changed, 1 insertion(+) diff --git a/library/train_util.py b/library/train_util.py index df6e24e2..d80ef516 100644 --- a/library/train_util.py +++ b/library/train_util.py @@ -842,6 +842,7 @@ class FineTuningDataset(BaseDataset): self.num_train_images = len(metadata) * dataset_repeats self.num_reg_images = 0 + # TODO do not record tag freq when no tag self.set_tag_frequency(os.path.basename(json_file_name), tags_list) self.dataset_dirs_info[os.path.basename(json_file_name)] = {"n_repeats": dataset_repeats, "img_count": len(metadata)}