Skip to content

Commit 0858858

Browse files
committed
Updated stabilization threshold
1 parent 31f34e7 commit 0858858

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
@@ -543,7 +543,7 @@ def solver_sinkhorn_stabilized_sparse(
543543

544544
rho_s, rho_t, eps = uot_params
545545
niters, tol, eval_freq = train_params
546-
tau = 1e3 # Threshold for stabilization
546+
tau = 1e6 # Threshold for stabilization
547547

548548
# Set up sparse matrix operations
549549
crow_indices = cost.crow_indices()

0 commit comments

Comments
 (0)