From c1a66317cda6000a5fe6303533ea418dbb2865aa Mon Sep 17 00:00:00 2001 From: Joel Holdbrooks Date: Mon, 23 Jan 2023 11:16:34 -0800 Subject: [PATCH] Forgot to set prepared_train_data --- data/data_loader.py | 1 + 1 file changed, 1 insertion(+) diff --git a/data/data_loader.py b/data/data_loader.py index 139ea3f..0df4e82 100644 --- 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)