`diffengine.models.losses.base`¶

Module Contents¶

Classes¶

Base class for all losses.

class diffengine.models.losses.base.BaseLoss(*args, **kwargs)[source]¶

Bases: torch.nn.Module

Base class for all losses.

property use_snr: bool[source]¶

Whether or not this loss uses SNR.

Return type:: bool