loss

_loss

This file is a base class for loss functions.

class scalr.nn.loss._loss.CustomLossBase[source]

Bases: Module

Base class to implement custom loss functions.

forward(out, preds)[source]: Returns loss betwen outputs and predictions.

scalr.nn.loss._loss.build_loss_fn(loss_config)[source]: Builder object to get Loss function, updated loss_config.