Loss Functions

class marius.nn.LossFunction

__init__(self: marius._nn.LossFunction) → None

class marius.nn.SoftmaxCrossEntropy

__init__(self: marius._nn.SoftmaxCrossEntropy, reduction: str = 'sum') → None

class marius.nn.RankingLoss

__init__(self: marius._nn.RankingLoss, reduction: str = 'sum', margin: float = 1.0) → None

class marius.nn.CrossEntropyLoss

__init__(self: marius._nn.CrossEntropyLoss, reduction: str = 'sum') → None

class marius.nn.BCEAfterSigmoidLoss

__init__(self: marius._nn.BCEAfterSigmoidLoss, reduction: str = 'sum') → None

class marius.nn.BCEWithLogitsLoss

__init__(self: marius._nn.BCEWithLogitsLoss, reduction: str = 'sum') → None

class marius.nn.MSELoss

__init__(self: marius._nn.MSELoss, reduction: str = 'sum') → None

class marius.nn.SoftPlusLoss

__init__(self: marius._nn.SoftPlusLoss, reduction: str = 'sum') → None