tomodrgn.beta_schedule.CyclicalSchedule#

class CyclicalSchedule(n_iterations: int, start_value: int = 0, end_value: int = 1, n_cycles: int = 4, fraction_cycle_increase: float = 0.5, increase_function: Literal['linear'] = 'linear')[source]#

Bases: object

Return a cyclically scaling beta value as a function of iteration. Implementation of https://doi.org/10.48550/arXiv.1903.10145, haofuml/cyclical_annealing

Methods

__call__(iteration)[source]#

Call self as a function.