modelzoo.vision.pytorch.dit.layers.schedulers#

Functions

get_beta_schedule

This is the deprecated API for creating beta schedules.

get_named_beta_schedule

Get a pre-defined beta schedule for the given name.