ertk.pytorch.utils.get_loss

ertk.pytorch.utils.get_loss(loss: str, *args, **kwargs) → Module

Get torch Module for activation function.

Parameters:

loss: str

String representing the loss function.

*args, **kwargs: optional

Arguments and keyword arguments to pass to loss function module __init__().

Returns:

torch.nn.Module

The loss function as a Module.