diff options
Diffstat (limited to 'ot/smooth.py')
| -rw-r--r-- | ot/smooth.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/ot/smooth.py b/ot/smooth.py index dd734b3..87d61ef 100644 --- a/ot/smooth.py +++ b/ot/smooth.py @@ -580,7 +580,7 @@ def smooth_ot_semi_dual(a, b, M, reg, reg_type='l2', method="L-BFGS-B", stopThr= raise NotImplementedError('Unknown regularization') # solve dual - alpha, res = solve_semi_dual(a, b, M, regul, max_iter=numItermax, + alpha, res = solve_semi_dual(a, b, M, regul, max_iter=numItermax, tol=stopThr, verbose=verbose) # reconstruct transport matrix |