diffengine.models.losses.utils¶
Module Contents¶
Functions¶
|
Compute SNR. |
- diffengine.models.losses.utils.compute_snr(timesteps, alphas_cumprod)[source]¶
Compute SNR.
Refer to https://github.com/TiankaiHang/Min-SNR-Diffusion-Tra ining/blob/521b624bd70c67cee4bdf49225915f5945a872e3/guided_diffusion/gaussi an_diffusion.py#L847-L849.
- Return type:
torch.Tensor