diffalign.models.epsnet.diffusion.linear_beta_schedule

diffalign.models.epsnet.diffusion.linear_beta_schedule(num_timesteps: int, beta_start: float = 0.0001, beta_end: float = 0.02) Tensor[source]