wassname/pytorch-lightning
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-17 12:40:36 +08:00
Code Issues Packages Projects Releases Wiki Activity

validation and training loops run the partial dataset (#1192)

Browse Source 
* auto_add_sampler() fix

* auto_add_sampler() fix

Co-authored-by: seth <seth@duckpapa.com>
This commit is contained in:
sneiman
2020-03-30 12:13:34 -04:00
committed by GitHub
co-authored by seth
parent 1d1aba812b
commit 6dfe9951e1
1 changed files with 3 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pytorch_lightning/trainer/data_loading.py
+3 -6
View File
@@ -91,15 +91,12 @@ class TrainerDataLoadingMixin(ABC):
)
dl_args['shuffle'] = False
else:
if train:
sampler = DistributedSampler(dataloader.dataset)
dl_args['shuffle'] = False
else:
sampler = SequentialSampler(dataloader.dataset)
sampler = DistributedSampler(dataloader.dataset)
dl_args['shuffle'] = False
dl_args['sampler'] = sampler
dataloader = DataLoader(**dl_args)
return dataloader
def reset_train_dataloader(self, model: LightningModule) -> None: