early stop starts counting once min epochs met

2019-07-16 10:00:03 -04:00 · 2019-07-16 10:00:03 -04:00 · 967e57f071
parent d12f6b7dd8
commit 967e57f071
1 changed files with 2 additions and 2 deletions
--- a/pytorch_lightning/models/trainer.py
+++ b/pytorch_lightning/models/trainer.py
@ -566,9 +566,9 @@ class Trainer(TrainerIO):
                model.on_epoch_end()

            # early stopping
-            if self.enable_early_stop:
+            met_min_epochs = epoch_nb > self.min_nb_epochs
+            if self.enable_early_stop and met_min_epochs:
                should_stop = self.early_stop_callback.on_epoch_end(epoch=epoch_nb, logs=self.__tng_tqdm_dic)
-                met_min_epochs = epoch_nb > self.min_nb_epochs

                # stop training
                stop = should_stop and met_min_epochs