ott.solvers.linear.continuous_barycenter.FreeBarycenterState.update

ott.solvers.linear.continuous_barycenter.FreeBarycenterState.update#

FreeBarycenterState.update(iteration, bar_prob, linear_ot_solver, store_errors)[source]#

Update the state of the solver.

Parameters:
  • iteration (int) – the current iteration of the outer loop.

  • bar_prob (FreeBarycenterProblem) – the barycenter problem.

  • linear_ot_solver (Any) – the linear OT solver to use.

  • store_errors (bool) – whether to store the errors of the inner loop.

Return type:

FreeBarycenterState

Returns:

The updated state.