diff --git a/pytorch_lightning/trainer/data_loading.py b/pytorch_lightning/trainer/data_loading.py index 00b37dfa..52e53acd 100644 --- a/pytorch_lightning/trainer/data_loading.py +++ b/pytorch_lightning/trainer/data_loading.py @@ -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