Mirrors
/
EveryDream2trainer
mirror of https://github.com/victorchall/EveryDream2trainer.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Forgot to set prepared_train_data

This commit is contained in:
Joel Holdbrooks 2023-01-23 11:16:34 -08:00
parent b6c7299baf
commit c1a66317cd
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
data/data_loader.py
View File

@ -166,6 +166,7 @@ class DataLoaderMultiAspect():
print (f" * DLMA: {len(items)} images loaded from {len(image_paths)} files") print (f" * DLMA: {len(items)} images loaded from {len(image_paths)} files")
self.prepared_train_data = items
random.Random(self.seed).shuffle(self.prepared_train_data) random.Random(self.seed).shuffle(self.prepared_train_data)
self.__report_undersized_images(events) self.__report_undersized_images(events)