Faster termination when training

2021-09-30 00:08:53 +02:00 · 2021-09-30 00:08:53 +02:00 · ca7b2fd87c
commit ca7b2fd87c
parent 1bd0597291
1 changed files with 1 additions and 1 deletions
--- 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-05 or best_mse > 3:
+    while gamma > 3.0e-06 and delta > 3.0e-05 or best_mse > 3:
        last_mse = mse
        print({'mse': mse, 'gamma': gamma, 'delta': delta})
        delta = sum(gradient[g]**2 for g in gradient)