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: