espnet2.tts.prodiff.denoiser.noise_scheduler

source

espnet2.tts.prodiff.denoiser.noise_scheduler(sched_type: str, timesteps: int, min_beta: float = 0.0, max_beta: float = 0.01, s: float = 0.008) → Tensor

Noise Scheduler.

• Parameters:
  • sched_type (str) – type of scheduler.
  • timesteps (int) – numbern of time steps.
  • min_beta (float , optional) – Minimum beta. Defaults to 0.0.
  • max_beta (float , optional) – Maximum beta. Defaults to 0.01.
  • s (float , optional) – Scheduler intersection. Defaults to 0.008.
• Returns: Noise.
• Return type: tensor