diff options
Diffstat (limited to 'ot/gpu/da.py')
-rw-r--r-- | ot/gpu/da.py | 2 |
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 |