Restrict Armijo linesearch function step to be at most 1 (PythonOT#189)

Thanks for the modification @panpan2
QianGuo123 · May 25, 2020 · 5af6452 · 5af6452
1 parent 0adc8aa
commit 5af6452
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ot/optim.py b/ot/optim.py
@@ -69,7 +69,7 @@ def phi(alpha1):
     alpha, phi1 = scalar_search_armijo(
         phi, phi0, derphi0, c1=c1, alpha0=alpha0)
 
-    return alpha, fc[0], phi1
+    return min(1, alpha), fc[0], phi1
 
 
 def solve_linesearch(cost, G, deltaG, Mi, f_val,