Skip to content

GitLab

Unverified Commit 8490f7e4 authored by dutxubo's avatar dutxubo Committed by GitHub
Browse files

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

parent 2f8524ba
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
nni/retiarii/oneshot/pytorch/random.py
View file @ 8490f7e4
...@@ -143,7 +143,8 @@ class SinglePathTrainer(BaseOneShotTrainer): ...@@ -143,7 +143,8 @@ class SinglePathTrainer(BaseOneShotTrainer):
self.train_loader = torch.utils.data.DataLoader(self.dataset_train, self.train_loader = torch.utils.data.DataLoader(self.dataset_train,
batch_size=batch_size, batch_size=batch_size,
num_workers=workers) num_workers=workers,
shuffle=True)
self.valid_loader = torch.utils.data.DataLoader(self.dataset_valid, self.valid_loader = torch.utils.data.DataLoader(self.dataset_valid,
batch_size=batch_size, batch_size=batch_size,
num_workers=workers) num_workers=workers)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment