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

fix _reset_eval_dataloader() for IterableDataset (#1560)

Browse Source 
* removed if dl from _reset_eval_dataloader()

* changed to if dl != None to be more safe

* hints from pep8speaks

Co-authored-by: ybrovman <ybrovman@ebay.com>
This commit is contained in:
Yuri Brovman
2020-05-05 14:09:48 -04:00
committed by GitHub
co-authored by ybrovman
parent fc7f5919b5
commit 35bbe178bd
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pytorch_lightning/trainer/data_loading.py
+1 -1
View File
@@ -219,7 +219,7 @@ class TrainerDataLoadingMixin(ABC):
' this off for validation and test dataloaders.')
# add samplers
dataloaders = [self.auto_add_sampler(dl, train=False) for dl in dataloaders if dl]
dataloaders = [self.auto_add_sampler(dl, train=False) for dl in dataloaders if dl is not None]
num_batches = 0