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."""