fixed default sampler (#1425)

2020-04-09 08:52:15 -04:00 · 2020-04-09 08:52:15 -04:00 · 21a1972921
parent b5c6d0e393
commit 21a1972921
1 changed files with 1 additions and 2 deletions
--- a/pytorch_lightning/trainer/data_loading.py
+++ b/pytorch_lightning/trainer/data_loading.py
@ -87,9 +87,8 @@ class TrainerDataLoadingMixin(ABC):
            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']