Scheduler

class flow_matching.path.scheduler.Scheduler

Base Scheduler class.

abstract 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