From 477b2549def08124e934bc0b4cb6f67a722c2037 Mon Sep 17 00:00:00 2001 From: Mike Clark Date: Wed, 3 Jun 2020 13:22:58 +0800 Subject: [PATCH] fix for #2051, always init with -inf The comparison `self.monitor_op == torch.gt` was always false, compare with mode instead for #2051 --- pytorch_lightning/callbacks/early_stopping.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pytorch_lightning/callbacks/early_stopping.py b/pytorch_lightning/callbacks/early_stopping.py index 61abfb87..63d9b86d 100644 --- a/pytorch_lightning/callbacks/early_stopping.py +++ b/pytorch_lightning/callbacks/early_stopping.py @@ -75,7 +75,7 @@ class EarlyStopping(Callback): if self.verbose > 0: log.info(f'EarlyStopping mode set to {self.mode} for monitoring {self.monitor}.') - self.min_delta *= 1 if self.monitor_op == torch.gt else -1 + self.min_delta *= 1 if self.mode == 'min' else -1 def _validate_condition_metric(self, logs): """