Skip to content

Commit 1754dba

Browse files
author
Pierre-Louis Barbarant
committed
Add docstring to epsilon scheduler in solver_sinkhorn_eps_scaling for clarity
1 parent 14a9fb4 commit 1754dba

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

src/fugw/solvers/utils.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -723,6 +723,7 @@ def solver_sinkhorn_eps_scaling(
723723
idx = 0
724724

725725
def get_reg(idx):
726+
"""Epsilon scheduler"""
726727
return (epsilon0 - eps) * torch.exp(
727728
-torch.tensor(idx, dtype=torch.float64)
728729
) + eps

0 commit comments

Comments
 (0)