diff options
| author | Alexandre Gramfort <alexandre.gramfort@m4x.org> | 2019-07-09 17:08:58 +0200 |
|---|---|---|
| committer | Alexandre Gramfort <alexandre.gramfort@m4x.org> | 2019-07-09 17:08:58 +0200 |
| commit | b6fb14861accd20a323bfc5ef96c20883e4f6ce1 (patch) | |
| tree | 3b3808e9fc6420822be2ffca81827c42d9c54d09 /ot/unbalanced.py | |
| parent | 1b00740a39f90f1e0bc7dc3a35723560c9ab4e97 (diff) | |
more
Diffstat (limited to 'ot/unbalanced.py')
| -rw-r--r-- | ot/unbalanced.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/ot/unbalanced.py b/ot/unbalanced.py index 50ec03c..467fda2 100644 --- a/ot/unbalanced.py +++ b/ot/unbalanced.py @@ -380,7 +380,8 @@ def sinkhorn_knopp_unbalanced(a, b, M, reg, alpha, numItermax=1000, print( '{:5s}|{:12s}'.format('It.', 'Err') + '\n' + '-' * 19) print('{:5d}|{:8e}|'.format(cpt, err)) - cpt = cpt + 1 + cpt += 1 + if log: log['u'] = u log['v'] = v |