diff options
Diffstat (limited to 'ot/smooth.py')
-rw-r--r-- | ot/smooth.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/ot/smooth.py b/ot/smooth.py index 4c46e73..dd734b3 100644 --- a/ot/smooth.py +++ b/ot/smooth.py @@ -580,10 +580,11 @@ 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, tol=stopThr) + alpha, res = solve_semi_dual(a, b, M, regul, max_iter=numItermax, + tol=stopThr, verbose=verbose) # reconstruct transport matrix - G = get_plan_from_semi_dual(alpha, b, M, regul, verbose=verbose) + G = get_plan_from_semi_dual(alpha, b, M, regul) if log: log = {'alpha': alpha, 'res': res} |