mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-10-29 13:30:57 +08:00
fix _reset_eval_dataloader() for IterableDataset (#1560)
* 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:
1 parent
fc7f5919b5
commit
35bbe178bd
1 file changed
+1
-1
@@ -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
|
||||
|
||||
|
||||
Reference in new issue
Block a user