Shuffle train dataloader (#42)

openvinotoolkit · Dec 21, 2021 · 5c9f9a9 · 5c9f9a9
1 parent f14d195
commit 5c9f9a9
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/anomalib/data/mvtec.py b/anomalib/data/mvtec.py
@@ -460,9 +460,7 @@ def setup(self, stage: Optional[str] = None) -> None:
 
     def train_dataloader(self) -> DataLoader:
         """Get train dataloader."""
-        return DataLoader(
-            self.train_data, shuffle=False, batch_size=self.train_batch_size, num_workers=self.num_workers
-        )
+        return DataLoader(self.train_data, shuffle=True, batch_size=self.train_batch_size, num_workers=self.num_workers)
 
     def val_dataloader(self) -> DataLoader:
         """Get validation dataloader."""