starling.data.schedulers.linear_beta_schedule

linear_beta_schedule(timesteps: int) → Tensor[source]

The beta values are linearly spaced between 0.0001 and 0.02.

Parameters:: timesteps (int) – The number of timesteps which will be used to generate the beta values.
Returns:: A tensor containing the beta values.
Return type:: torch.Tensor