summaryrefslogtreecommitdiff
path: root/ot/gromov.py
diff options
context:
space:
mode:
authorRémi Flamary <remi.flamary@gmail.com>2018-02-16 14:08:55 +0100
committerRémi Flamary <remi.flamary@gmail.com>2018-02-16 14:08:55 +0100
commitb4665fe3c12780af4228cb1fb7dc8e1159c81f63 (patch)
tree46e11035c4432acc27777e745bfd75b8ad7f7372 /ot/gromov.py
parentd41ffdb24de5cb234237482b3f332b06514b10f0 (diff)
should pass tests now
Diffstat (limited to 'ot/gromov.py')
-rw-r--r--ot/gromov.py2
1 files changed, 2 insertions, 0 deletions
diff --git a/ot/gromov.py b/ot/gromov.py
index b1e9ee0..2a23873 100644
--- a/ot/gromov.py
+++ b/ot/gromov.py
@@ -307,6 +307,8 @@ def gromov_wasserstein(C1, C2, p, q, loss_fun, log=False, **kwargs):
Print information along iterations
log : bool, optional
record log if True
+ **kwargs : dict
+ parameters can be directly pased to the ot.optim.cg solver
Returns
-------