osc.train.ModelLoss

class ModelLoss(*args, **kwargs)[source]

Bases: Protocol

Function type signature for model loss functions.

Methods

__call__(output, *, reduction='mean')[source]

Call self as a function.

Return type: Optional[Tensor]