データセットにepoch、stepが通達されないバグ修正

2026-04-08 22:35:09 +00:00 · 2023-03-26 01:44:25 +09:00
parent 1b89b2a10e
commit 292cdb8379
3 changed files with 24 additions and 11 deletions
--- a/library/train_util.py
+++ b/library/train_util.py
@@ -517,6 +517,7 @@ class BaseDataset(torch.utils.data.Dataset):
                else:
                    caption = caption.replace(str_from, str_to)

+        print(self.current_step, self.max_train_steps, caption)
        return caption

    def get_input_ids(self, caption):