Skip to content

Commit 39993be

Browse files
author
Pierre-Louis Barbarant
committed
Change tensor dtype from float32 to float64 in epsilon scheduler for improved precision
1 parent 1754dba commit 39993be

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/fugw/solvers/utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -804,7 +804,7 @@ def solver_sinkhorn_eps_scaling_sparse(
804804
def get_reg(idx):
805805
"""Epsilon scheduler"""
806806
return (epsilon0 - eps) * torch.exp(
807-
-torch.tensor(idx, dtype=torch.float32)
807+
-torch.tensor(idx, dtype=torch.float64)
808808
) + eps
809809

810810
numItermin = 35

0 commit comments

Comments
 (0)