tomodrgn.beta_schedule

Classes and functions to facilitate sampling beta values per minibatch when training a VAE.

Functions

get_beta_schedule

Return one of several types of beta schedules: Constant, Linear, or Cyclical.

Classes

ConstantSchedule	Return a constant value for beta, independent of iteration.
CyclicalSchedule	Return a cyclically scaling beta value as a function of iteration.
LinearSchedule	Return a linearly scaling beta value as a function of iteration.