Class RescorerLoss

Inheritance Relationships

Base Type

Class Documentation

class RescorerLoss : public marian::CrossEntropyLoss

Cross entropy in rescorer used for computing sentences-level or word-level log probabilities.

This class differs from CrossEntropy in the different ‘axes’ setting, and that label smoothing is disabled.

Public Functions

RescorerLoss(bool wordScores)
virtual RationalLoss apply(Logits logits, const Words &labels, Expr mask = nullptr, Expr labelWeights = nullptr)