modelzoo.vision.pytorch.dit.layers.schedulers.get_named_beta_schedule#
- modelzoo.vision.pytorch.dit.layers.schedulers.get_named_beta_schedule(schedule_name, num_diffusion_timesteps, beta_start=0.0001, beta_end=0.02)[source]#
Get a pre-defined beta schedule for the given name. The beta schedule library consists of beta schedules which remain similar in the limit of num_diffusion_timesteps. Beta schedules may be added, but should not be removed or changed once they are committed to maintain backwards compatibility.