Filter out items with errors

2023-01-24 08:33:20 -08:00 · 2023-01-24 08:33:20 -08:00 · e6d1096637
parent f0e8ee55df
commit e6d1096637
1 changed files with 1 additions and 1 deletions
--- a/data/data_loader.py
+++ b/data/data_loader.py
@ -166,7 +166,7 @@ class DataLoaderMultiAspect():

            print (f" * DLMA: {len(items)} images loaded from {len(image_paths)} files")
            
-            self.prepared_train_data = items
+            self.prepared_train_data = [item for item in items if item.error is None]
            random.Random(self.seed).shuffle(self.prepared_train_data)
            self.__report_errors(items)