L2Regularization
- class lightning_ir.loss.loss.L2Regularization(query_weight: float = 0.0001, doc_weight: float = 0.0001)[source]
Bases:
RegularizationLossFunction
Methods
__init__
([query_weight, doc_weight])compute_loss
(query_embeddings, doc_embeddings)