Update early_stopping.py

Lightning-AI · May 25, 2020 · 1680c88 · 1680c88
1 parent 0a954a3
commit 1680c88
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/pytorch_lightning/callbacks/early_stopping.py b/pytorch_lightning/callbacks/early_stopping.py
@@ -110,6 +110,9 @@ def on_train_start(self, trainer, pl_module):
         self.best = torch_inf if self.monitor_op == torch.lt else -torch_inf
 
     def on_validation_end(self, trainer, pl_module):
+        self._run_early_stopping_check(trainer, pl_module)
+
+    def _run_early_stopping_check(self, trainer, pl_module):
         logs = trainer.callback_metrics
         stop_training = False
         if not self._validate_condition_metric(logs):