ott.tools.progot.get_alpha_schedule#
- ott.tools.progot.get_alpha_schedule(kind, *, num_steps)[source]#
Get the step size schedule.
Convenience wrapper to get a sequence of
num_steps
timestamps between 0 and 1, distributed according to thekind
option below. See Section 4 in [Kassraie et al., 2024] for more details.