diff --git a/pytorch_lightning/trainer/data_loading.py b/pytorch_lightning/trainer/data_loading.py index 66b83fd4bfd005..50e86fd5c7ad2f 100644 --- a/pytorch_lightning/trainer/data_loading.py +++ b/pytorch_lightning/trainer/data_loading.py @@ -87,9 +87,8 @@ def auto_add_sampler(self, dataloader: DataLoader, train: bool) -> DataLoader: return dataloader need_dist_sampler = self.use_ddp or self.use_ddp2 or self.use_tpu - no_sampler_added = dataloader.sampler is None - if need_dist_sampler and no_sampler_added: + if need_dist_sampler: skip_keys = ['sampler', 'batch_sampler', 'dataset_kind']