Class RescorerLoss

• Defined in File loss.h

Inheritance Relationships

Base Type

• public marian::CrossEntropyLoss (Class CrossEntropyLoss)

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)