validation and training loops run the partial dataset (#1192)

* auto_add_sampler() fix

* auto_add_sampler() fix

Co-authored-by: seth <seth@duckpapa.com>
This commit is contained in:
sneiman 2020-03-30 09:13:34 -07:00 committed by GitHub
parent 1d1aba812b
commit 6dfe9951e1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 6 deletions

View File

@ -91,15 +91,12 @@ class TrainerDataLoadingMixin(ABC):
)
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: