diff --git a/pytorch_lightning/trainer/training_loop.py b/pytorch_lightning/trainer/training_loop.py index cbe11864..576ecfb5 100644 --- a/pytorch_lightning/trainer/training_loop.py +++ b/pytorch_lightning/trainer/training_loop.py @@ -347,13 +347,13 @@ class TrainerTrainLoopMixin(ABC): # ----------------- self.run_training_epoch() - # update LR schedulers - self.update_learning_rates(interval='epoch') - if self.max_steps and self.max_steps == self.global_step: self.run_training_teardown() return + # update LR schedulers + self.update_learning_rates(interval='epoch') + # early stopping met_min_epochs = epoch >= self.min_epochs - 1 met_min_steps = self.global_step >= self.min_steps if self.min_steps else True