diff options
| author | Leo gautheron <gautheron@iv-cm-359.creatis.insa-lyon.fr> | 2017-04-20 13:51:30 +0200 |
|---|---|---|
| committer | Leo gautheron <gautheron@iv-cm-359.creatis.insa-lyon.fr> | 2017-04-20 13:51:30 +0200 |
| commit | 67f09ad4089594457cb00702dd3aa8a6a94ec2eb (patch) | |
| tree | d458e5661f8ae9dd1dfe8127003ac4141f74b7d0 /ot/gpu/bregman.py | |
| parent | 16f51f971607efab2c73958d207c582b389406c8 (diff) | |
changes from feedback
Diffstat (limited to 'ot/gpu/bregman.py')
| -rw-r--r-- | ot/gpu/bregman.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/ot/gpu/bregman.py b/ot/gpu/bregman.py index 3fdf11b..4d4a8b7 100644 --- a/ot/gpu/bregman.py +++ b/ot/gpu/bregman.py @@ -6,7 +6,7 @@ Bregman projections for regularized OT with GPU import numpy as np -def sinkhornGPU(a, b, M_GPU, reg, numItermax=1000, stopThr=1e-9, verbose=False, +def sinkhorn(a, b, M_GPU, reg, numItermax=1000, stopThr=1e-9, verbose=False, log=False, cudamat=None): # init data Nini = len(a) |