Skip to content

Commit 5b28e5f

Browse files
committed
uncommited ground bar fix
1 parent ea5feca commit 5b28e5f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

ot/lp/_barycenter_solvers.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -696,7 +696,7 @@ def ground_bary(y, x_init):
696696
if i == 0:
697697
diff = ground_bary_stopThr + 1.0
698698
else:
699-
diff = (loss.item() - loss_prev) ** 2
699+
diff = loss_prev - loss.item()
700700
loss_prev = loss.item()
701701
if diff < ground_bary_stopThr:
702702
break

0 commit comments

Comments
 (0)