MarginMSE

class lightning_ir.loss.loss.MarginMSE(margin: float | Literal['scores'] = 1.0)

Bases: PairwiseLossFunction

__init__(margin: float | Literal['scores'] = 1.0)

Methods

__init__([margin])
compute_loss(output, batch)