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.

previous

modelzoo.vision.pytorch.dit.layers.TimestepEmbeddingLayer.TimestepEmbeddingLayer

next

modelzoo.vision.pytorch.dit.layers.schedulers.get_beta_schedule