Skip to contents

Beta schedule values for T diffusion steps (hmdfw)

Usage

morie_beta_schedule_values(
  T,
  beta_schedule = "linear",
  beta_start = 1e-04,
  beta_end = 0.02
)

Arguments

T

Number of steps.

beta_schedule

"linear", "cosine", or an explicit length-T vector.

beta_start, beta_end

Linear-schedule endpoints.

Value

Numeric vector of length T.

Examples

morie_beta_schedule_values(T = 8L)
#> [1] 0.000100000 0.002942857 0.005785714 0.008628571 0.011471429 0.014314286
#> [7] 0.017157143 0.020000000
morie_beta_schedule_values(T = 8L, beta_schedule = "cosine")
#> [1] 0.04219541 0.11567331 0.19218803 0.27824552 0.38362086 0.52603638 0.74026888
#> [8] 0.99900000