Skip to content

Commit d370f18

Browse files
committed
bug verbose semi-dual
1 parent fb883fc commit d370f18

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

ot/smooth.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -580,10 +580,11 @@ def smooth_ot_semi_dual(a, b, M, reg, reg_type='l2', method="L-BFGS-B", stopThr=
580580
raise NotImplementedError('Unknown regularization')
581581

582582
# solve dual
583-
alpha, res = solve_semi_dual(a, b, M, regul, max_iter=numItermax, tol=stopThr)
583+
alpha, res = solve_semi_dual(a, b, M, regul, max_iter=numItermax,
584+
tol=stopThr, verbose=verbose)
584585

585586
# reconstruct transport matrix
586-
G = get_plan_from_semi_dual(alpha, b, M, regul, verbose=verbose)
587+
G = get_plan_from_semi_dual(alpha, b, M, regul)
587588

588589
if log:
589590
log = {'alpha': alpha, 'res': res}

0 commit comments

Comments
 (0)