diff --git a/declearn/main/utils/_early_stop.py b/declearn/main/utils/_early_stop.py index 92155e4832127725a7845f3b1b625d112ea814bd..f703a9b95422733ce1e4fa93dbc8b49b91e49a01 100644 --- a/declearn/main/utils/_early_stop.py +++ b/declearn/main/utils/_early_stop.py @@ -105,7 +105,7 @@ class EarlyStopping: self._best_metric = metric if self.relative: diff /= self._best_metric - if diff < self.tolerance: + if diff <= self.tolerance: self._n_iter_stuck += 1 else: self._n_iter_stuck = 0