From 6c9b983d49b6b5ea76a4cf53b6ce6df7671765fd Mon Sep 17 00:00:00 2001
From: Paul Andrey <paul.andrey@inria.fr>
Date: Wed, 30 Aug 2023 15:02:13 +0200
Subject: [PATCH] Fix EarlyStopping for exact repeated inputs.
---
declearn/main/utils/_early_stop.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/declearn/main/utils/_early_stop.py b/declearn/main/utils/_early_stop.py
index 92155e48..f703a9b9 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
--
GitLab