From 21a1972921809ea04ab1e4c657e326dfecd5e352 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Thu, 9 Apr 2020 08:52:15 -0400 Subject: [PATCH] fixed default sampler (#1425) --- pytorch_lightning/trainer/data_loading.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/pytorch_lightning/trainer/data_loading.py b/pytorch_lightning/trainer/data_loading.py index 66b83fd4bf..50e86fd5c7 100644 --- 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']