InfoNCE

class lightning_ir.loss.loss.InfoNCE[source]

Bases: ListwiseLossFunction

__init__()

Methods

compute_loss(output, batch)