VPScheduler

class flow_matching.path.scheduler.VPScheduler(beta_min: float = 0.1, beta_max: float = 20.0)

Variance Preserving Scheduler.

snr_inverse(snr: Tensor) → Tensor

Computes \(t\) from the signal-to-noise ratio \(\frac{\alpha_t}{\sigma_t}\).

Parameters:

snr (Tensor) – The signal-to-noise, shape (…)

Returns:

t, shape (…)

Return type:

Tensor