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

* auto_add_sampler() fix * auto_add_sampler() fix Co-authored-by: seth <seth@duckpapa.com>
Lightning-AI · Mar 30, 2020 · 6dfe995 · 6dfe995
1 parent 1d1aba8
commit 6dfe995
Showing 1 changed file with 3 additions and 6 deletions.
diff --git 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: