From 0fac2d64cf4c6c31d3b21605a272ace01d0c51a2 Mon Sep 17 00:00:00 2001 From: Hiroyuki Vincent Yamazaki Date: Fri, 18 Oct 2019 17:18:05 +0900 Subject: [PATCH] Fix off-by-one epoch length (#377) --- 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 d352052a..d7a3ca01 100644 --- a/pytorch_lightning/trainer/trainer.py +++ b/pytorch_lightning/trainer/trainer.py @@ -1128,7 +1128,7 @@ class Trainer(TrainerIOMixin): # stop when the flag is changed or we've gone past the amount # requested in the batches self.total_batch_nb += 1 - met_batch_limit = batch_nb > self.nb_training_batches + met_batch_limit = batch_nb >= self.nb_training_batches if met_batch_limit: break