diff options
author | ncassereau-idris <84033440+ncassereau-idris@users.noreply.github.com> | 2021-09-30 08:36:24 +0200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2021-09-30 08:36:24 +0200 |
commit | 14c30d4cfac060ff0bf8c64d4c88c77df32aad86 (patch) | |
tree | 082e4b920a3cd34a29ebb3c6e6500af9f0a7fdcc /ot/bregman.py | |
parent | 1c7e7ce2da8bb362c184fb6eae71fe7e36356494 (diff) |
[MRG] GPU bugs solve (#288)
* gpus tests now passing
* pep8 compliance
* GPU tests succeeding even if b has rank higher than 1
Co-authored-by: RĂ©mi Flamary <remi.flamary@gmail.com>
Diffstat (limited to 'ot/bregman.py')
-rw-r--r-- | ot/bregman.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/ot/bregman.py b/ot/bregman.py index 105b38b..317c902 100644 --- a/ot/bregman.py +++ b/ot/bregman.py @@ -312,7 +312,7 @@ def sinkhorn_knopp(a, b, M, reg, numItermax=1000, numItermax : int, optional Max number of iterations stopThr : float, optional - Stop threshol on error (>0) + Stop threshold on error (>0) verbose : bool, optional Print information along iterations log : bool, optional |