From d375e2adc5906a41a7a2a9ed76f03f7a4d2d2681 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Adrian=20W=C3=A4lchli?= Date: Sat, 20 Jun 2020 12:13:40 +0200 Subject: [PATCH] fix list size --- pytorch_lightning/trainer/trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pytorch_lightning/trainer/trainer.py b/pytorch_lightning/trainer/trainer.py index 166a479499964..6e789f2a12aa5 100644 --- a/pytorch_lightning/trainer/trainer.py +++ b/pytorch_lightning/trainer/trainer.py @@ -1070,7 +1070,7 @@ def run_pretrain_routine(self, model: LightningModule): num_loaders = len(self.val_dataloaders) max_batches = [self.num_sanity_val_steps] * num_loaders - max_batches = [float('inf') for m in max_batches if m == -1] + max_batches = [float('inf') if m == -1 else m for m in max_batches] eval_results = self._evaluate(model, self.val_dataloaders, max_batches,