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)