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

__init__([query_weight, doc_weight])

compute_loss(output)

process_embeddings(output)

process_scores(output)

process_targets(scores, batch)