diffalign.models.epsnet.diffusion.cosine_beta_schedule¶

diffalign.models.epsnet.diffusion.cosine_beta_schedule(num_timesteps: int, s: float = 0.008) → Tensor[source]¶: Nichol & Dhariwal (2021): https://arxiv.org/abs/2102.09672 Returns betas of length T (float32).