nami.schedules.edm# Classes EDMSchedule([sigma_min, sigma_max, rho]) class nami.schedules.edm.EDMSchedule(sigma_min=0.002, sigma_max=80.0, rho=7.0)[source]# Bases: NoiseSchedule Parameters: sigma_min (float) sigma_max (float) rho (float) alpha(t)[source]# Return type: Tensor Parameters: t (Tensor) diffusion(t)[source]# Return type: Tensor Parameters: t (Tensor) drift(x, t)[source]# Return type: Tensor Parameters: x (Tensor) t (Tensor) sigma(t)[source]# Return type: Tensor Parameters: t (Tensor)
