From 8964fa2c6a444cb3ee04ed5c574c411844c0ab86 Mon Sep 17 00:00:00 2001 From: Dominik Roth Date: Sun, 26 Sep 2021 23:18:21 +0200 Subject: [PATCH] No early stopping if still bad --- caliGraph.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/caliGraph.py b/caliGraph.py index dc4b5a7..718ac97 100755 --- a/caliGraph.py +++ b/caliGraph.py @@ -835,7 +835,7 @@ def train(gamma = 1, full=True): best_mse = mse stagLen = 0 - while gamma > 1.0e-06 and delta > 1.0e-06: + while gamma > 1.0e-06 and delta > 1.0e-06 or best_mse > 3: last_mse = mse print({'mse': mse, 'gamma': gamma, 'delta': delta}) delta = sum(gradient[g]**2 for g in gradient)