summaryrefslogtreecommitdiff
path: root/ot/gpu
diff options
context:
space:
mode:
authorLeo gautheron <gautheron@iv-cm-359.creatis.insa-lyon.fr>2017-04-20 14:06:17 +0200
committerLeo gautheron <gautheron@iv-cm-359.creatis.insa-lyon.fr>2017-04-20 14:06:17 +0200
commit9c5cc82b2e6f6dcd94a24a47f414c278c717ee7a (patch)
tree4c3b6ec936222bedc51bea7768cfb3a62729a1ee /ot/gpu
parent67f09ad4089594457cb00702dd3aa8a6a94ec2eb (diff)
missing argument function sinkhorn
Diffstat (limited to 'ot/gpu')
-rw-r--r--ot/gpu/da.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/ot/gpu/da.py b/ot/gpu/da.py
index 8b45772..4c583a7 100644
--- a/ot/gpu/da.py
+++ b/ot/gpu/da.py
@@ -77,5 +77,5 @@ class OTDA_sinkhorn_GPU(OTDA):
M = np.log(1 + np.log(1 + self.M_GPU.asarray()))
self.M_GPU = cudamat.CUDAMatrix(M)
- self.G = sinkhorn(ws, wt, self.M_GPU, reg, cudamat=cudamat)
+ self.G = sinkhorn(ws, wt, self.M_GPU, reg, cudamat=cudamat, **kwargs)
self.computed = True