validation and training loops run the partial dataset (Lightning-AI#1192

) * auto_add_sampler() fix * auto_add_sampler() fix Co-authored-by: seth <seth@duckpapa.com>
alexeykarnachev · Apr 3, 2020 · f9a2b75 · f9a2b75
1 parent e4a8751
commit f9a2b75
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: