helios.losses.utils¶

Attributes¶

LOSS_REGISTRY

Global instance of the registry for loss functions.

Functions¶

create_loss(→ torch.nn.Module)

Create the loss function for the given type.

Module Contents¶

helios.losses.utils.LOSS_REGISTRY¶

Global instance of the registry for loss functions.

Example

import helios.losses as hll

# This automatically registers your loss function.
@hll.LOSS_REGISTRY.register
class MyLoss:
    ...

# Alternatively you can manually register a loss function like this:
hll.LOSS_REGISTRY.register(MyLoss)

helios.losses.utils.create_loss(type_name: str, *args: Any, **kwargs: Any) → torch.nn.Module¶

Create the loss function for the given type.

Parameters:

type_name – the type of the loss to create.
args – positional arguments to pass into the loss.
kwargs – keyword arguments to pass into the loss.

Returns:

The loss function.