wassname/pytorch-lightning
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-11 12:31:23 +08:00
Code Issues Packages Projects Releases Wiki Activity

fixing bug in testing for IterableDataset (#547)

Browse Source
This commit is contained in:
MikeScarp
2019-11-26 04:59:20 -05:00
committed by William Falcon
parent 462788738b
commit 55f3ffd7c7
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pytorch_lightning/trainer/data_loading_mixin.py
+2 -2
View File
@@ -24,7 +24,7 @@ class TrainerDataLoadingMixin(object):
self.get_train_dataloader = model.train_dataloader
# determine number of training batches
if isinstance(self.get_train_dataloader(), IterableDataset):
if isinstance(self.get_train_dataloader().dataset, IterableDataset):
self.nb_training_batches = float('inf')
else:
self.nb_training_batches = len(self.get_train_dataloader())
@@ -167,7 +167,7 @@ class TrainerDataLoadingMixin(object):
self.get_val_dataloaders()
# support IterableDataset for train data
self.is_iterable_train_dataloader = isinstance(self.get_train_dataloader(), IterableDataset)
self.is_iterable_train_dataloader = isinstance(self.get_train_dataloader().dataset, IterableDataset)
if self.is_iterable_train_dataloader and not isinstance(self.val_check_interval, int):
m = '''
When using an iterableDataset for train_dataloader,