[Fix] shuffle is set to True in train_loader of SinglePathTrainer (#4696)

8490f7e4 · dutxubo · GitHub · 2f8524ba · 8490f7e4
Unverified Commit 8490f7e4 authored Mar 25, 2022 by dutxubo Committed by GitHub Mar 25, 2022
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

nni/retiarii/oneshot/pytorch/random.py nni/retiarii/oneshot/pytorch/random.py +2 -1

No files found.
--- a/nni/retiarii/oneshot/pytorch/random.py
+++ b/nni/retiarii/oneshot/pytorch/random.py
@@ -143,7 +143,8 @@ class SinglePathTrainer(BaseOneShotTrainer):

        self.train_loader = torch.utils.data.DataLoader(self.dataset_train,
                                                        batch_size=batch_size,
-                                                        num_workers=workers)
+                                                        num_workers=workers,
+                                                        shuffle=True)
        self.valid_loader = torch.utils.data.DataLoader(self.dataset_valid,
                                                        batch_size=batch_size,
                                                        num_workers=workers)