From 48e76c8aab3cca0b3f1735194118bcac39d9759e Mon Sep 17 00:00:00 2001
From: DKnight54 <126916963+DKnight54@users.noreply.github.com>
Date: Mon, 26 May 2025 02:05:43 +0800
Subject: [PATCH] Update train_util.py

---
 library/train_util.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/library/train_util.py b/library/train_util.py
index b958e189..05e21f83 100644
--- a/library/train_util.py
+++ b/library/train_util.py
@@ -1616,6 +1616,7 @@ class DreamBoothDataset(BaseDataset):
         self.latents_cache = None
         self.reg_infos: Dict[str, Tuple[ImageInfo, DreamBoothSubset]] = {}
         self.reg_infos_index: List[str] = []
+        self.reg_infos_index_traverser = 0
 
         self.enable_bucket = enable_bucket
         if self.enable_bucket:
@@ -1790,6 +1791,7 @@ class DreamBoothDataset(BaseDataset):
 
         logger.info(f"{num_reg_images} reg images.")
         self.num_reg_images = num_reg_images
+        self.reg_infos_index_traverser = 0
 
     def set_reg_randomize(self, reg_randomize = False):
         self.reg_randomize = reg_randomize