From e5d2ca532c96f743b99276971140536e1673d5c4 Mon Sep 17 00:00:00 2001 From: Dominik Roth Date: Sat, 22 Jan 2022 18:12:57 +0100 Subject: [PATCH] Shorter Training --- caliGraph.py | 1 + 1 file changed, 1 insertion(+) diff --git a/caliGraph.py b/caliGraph.py index 76af4fc..4c2577b 100755 --- a/caliGraph.py +++ b/caliGraph.py @@ -953,6 +953,7 @@ def train(initGamma, full=True): goal = 1.0e-5 while gamma > goal and delta > goal or best_mse > 15: + goal *= 1.1 last_mse = mse print({'mse': mse, 'gamma': gamma, 'delta': delta}) delta = sum(gradient[g]**2 for g in gradient)