torch_ecg.models.loss.setup_criterion

torch_ecg.models.loss.setup_criterion(name: str, **kwargs: Any) → Module

Setup the criterion (loss function).

Parameters:

• name (str) – The name of the criterion.

• **kwargs (Any) – The keyword arguments for the criterion.

Returns:

The criterion (loss function).

Return type:

nn.Module