scalr.nn.loss package

Submodules

scalr.nn.loss._loss module

This file is a base class for loss functions.

class scalr.nn.loss._loss.CustomLossBase

Bases: Module

Base class to implement custom loss functions.

forward(out, preds)

Returns loss betwen outputs and predictions.

scalr.nn.loss._loss.build_loss_fn(loss_config)

Builder object to get Loss function, updated loss_config.

Module contents