From 26d3f0dbea3ebb4d0db58f0f382fc633f275d067 Mon Sep 17 00:00:00 2001 From: sebftw Date: Mon, 19 Aug 2019 00:15:58 +0200 Subject: [PATCH] Error if dataset size = 1 batch. (#141) Fix for the bug mentioned in https://github.com/williamFalcon/pytorch-lightning/issues/139 --- pytorch_lightning/models/trainer.py | 1 + 1 file changed, 1 insertion(+) diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py index 3499911e..8ec6d785 100644 --- a/pytorch_lightning/models/trainer.py +++ b/pytorch_lightning/models/trainer.py @@ -370,6 +370,7 @@ class Trainer(TrainerIO): # determine when to check validation self.val_check_batch = int(self.nb_tng_batches * self.val_check_interval) + self.val_check_batch = max(1, self.val_check_batch) def __add_tqdm_metrics(self, metrics): for k, v in metrics.items():