log to print tag frequencies

2026-04-08 22:35:09 +00:00 · 2024-02-10 09:59:12 +09:00
parent 5d9e2873f6
commit 7202596393
1 changed files with 2 additions and 2 deletions
--- a/finetune/tag_images_by_wd14_tagger.py
+++ b/finetune/tag_images_by_wd14_tagger.py
@@ -291,9 +291,9 @@ def main(args):

    if args.frequency_tags:
        sorted_tags = sorted(tag_freq.items(), key=lambda x: x[1], reverse=True)
-        logger.info("Tag frequencies:")
+        print("Tag frequencies:")
        for tag, freq in sorted_tags:
-            logger.info(f"{tag}: {freq}")
+            print(f"{tag}: {freq}")

    logger.info("done!")