fix list size

Lightning-AI · Jun 20, 2020 · d375e2a · d375e2a
1 parent e1e9680
commit d375e2a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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,