jax_sgmc.scheduler.cyclic_burn_in

jax_sgmc.scheduler.cyclic_burn_in(beta=1.0, k=1)[source]

Discards samples at the beginning of each cycle.

Parameters:

beta (Array) – Ratio of current and initial step size up to which burn in should be applied
k (int) – Number of cycles

Returns:

Returns a burn in schedule, which applies burn in to the beginning of each cycle.