From 2d2407410edd6abab37c483cbf867fc00a801be6 Mon Sep 17 00:00:00 2001
From: Kohya S <ykumeykume@gmail.com>
Date: Thu, 2 Mar 2023 21:32:02 +0900
Subject: [PATCH] show index in caching latents

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

diff --git a/library/train_util.py b/library/train_util.py
index 7a285a94..75176e13 100644
--- a/library/train_util.py
+++ b/library/train_util.py
@@ -1061,7 +1061,8 @@ class DatasetGroup(torch.utils.data.ConcatDataset):
   #     dataset.make_buckets()
 
   def cache_latents(self, vae):
-    for dataset in self.datasets:
+    for i, dataset in enumerate(self.datasets):
+      print(f"[Dataset {i}]")
       dataset.cache_latents(vae)
 
   def is_latent_cacheable(self) -> bool: