FLOPSRegularization

class lightning_ir.loss.loss.FLOPSRegularization(query_weight: float = 0.0001, doc_weight: float = 0.0001)[source]

Bases: RegularizationLossFunction

__init__(query_weight: float = 0.0001, doc_weight: float = 0.0001) → None

Methods

compute_loss(output)