Forgot to set prepared_train_data

2023-01-23 11:16:34 -08:00 · 2023-01-23 11:16:34 -08:00 · c1a66317cd
parent b6c7299baf
commit c1a66317cd
1 changed files with 1 additions and 0 deletions
--- a/data/data_loader.py
+++ b/data/data_loader.py
@ -166,6 +166,7 @@ class DataLoaderMultiAspect():

            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)
            self.__report_undersized_images(events)