KLDivergence

class lightning_ir.loss.loss.KLDivergence[source]

Bases: ListwiseLossFunction

__init__()

Methods

compute_loss(output, batch)