Skip to content

Loss Factory

Source

Loss factory module.

This module contains a factory function for creating loss functions.

FCNLossParameters dataclass

Dataclass for FCN loss.

loss_factory(loss)

Return loss function.

Parameters:

Name Type Description Default
loss Union[SupportedLoss, str]

Requested loss.

required

Returns:

Type Description
Callable

Loss function.

Raises:

Type Description
NotImplementedError

If the loss is unknown.