- class ott.geometry.epsilon_scheduler.Epsilon(target=None, scale_epsilon=None, init=None, decay=None)#
Scheduler class for the regularization parameter epsilon.
An epsilon scheduler outputs a regularization strength, to be used by in a Sinkhorn-type algorithm, at any iteration count. That value is either the final, targeted regularization, or one that is larger, obtained by geometric decay of an initial value that is larger than the intended target. Concretely, the value returned by such a scheduler will consider first the max between
init * target * decay ** iteration. If the
scale_epsilonparameter is provided, that value is used to multiply the max computed previously by
Return (intermediate) regularizer value at a given iteration.
Create or return an Epsilon instance.
Return the final regularizer value of scheduler.