Fix shuffle in train dataloader

openvinotoolkit · Jul 3, 2023 · a70c521 · a70c521
1 parent 927cadc
commit a70c521
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/otx/algorithms/visual_prompting/adapters/pytorch_lightning/datasets/dataset.py b/otx/algorithms/visual_prompting/adapters/pytorch_lightning/datasets/dataset.py
@@ -282,7 +282,7 @@ def train_dataloader(self) -> Union[DataLoader, List[DataLoader], Dict[str, Data
         """
         return DataLoader(
             self.train_dataset,
-            shuffle=False,
+            shuffle=True,
             batch_size=self.config.train_batch_size,
             num_workers=self.config.num_workers,
             collate_fn=collate_fn,