Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fixed default sampler (#1425)

This commit is contained in:
William Falcon 2020-04-09 08:52:15 -04:00 committed by GitHub
parent b5c6d0e393
commit 21a1972921
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
pytorch_lightning/trainer/data_loading.py
View File

@ -87,9 +87,8 @@ class TrainerDataLoadingMixin(ABC):
return dataloader return dataloader
need_dist_sampler = self.use_ddp or self.use_ddp2 or self.use_tpu 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'] skip_keys = ['sampler', 'batch_sampler', 'dataset_kind']