Skip to content

Commit 8cc200f

Browse files
committed
update doc
1 parent 705e0a9 commit 8cc200f

1 file changed

Lines changed: 5 additions & 0 deletions

File tree

ot/bregman/_geomloss.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -133,6 +133,11 @@ def empirical_sinkhorn2_geomloss(
133133
better stability and epsilon-scaling. The solution is computed in a lazy way
134134
using the Geomloss [60]_ and the KeOps library [61]_.
135135
136+
..warning::
137+
The Geomloss library is required for this function to work. Also
138+
when setting `log=True`, the dual potentials are computed using autograd and
139+
may be slow for large problems and prevent computing backward gradients.
140+
136141
Parameters
137142
----------
138143
X_s : array-like, shape (n_samples_a, dim)

0 commit comments

Comments
 (0)