From f9a2b750315fd05da91ff30282f9573fb917e2ea Mon Sep 17 00:00:00 2001 From: sneiman Date: Mon, 30 Mar 2020 09:13:34 -0700 Subject: [PATCH] validation and training loops run the partial dataset (#1192) * auto_add_sampler() fix * auto_add_sampler() fix Co-authored-by: seth --- pytorch_lightning/trainer/data_loading.py | 9 +++------ 1 file changed, 3 insertions(+), 6 deletions(-) diff --git a/pytorch_lightning/trainer/data_loading.py b/pytorch_lightning/trainer/data_loading.py index dced28144fbe8..0ee13109de087 100644 --- a/pytorch_lightning/trainer/data_loading.py +++ b/pytorch_lightning/trainer/data_loading.py @@ -91,15 +91,12 @@ def auto_add_sampler(self, dataloader: DataLoader, train: bool) -> DataLoader: ) 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: