Expand description
Generic beam search infrastructure.
Re-exports§
pub use beam::BeamConfig;pub use beam::BeamSearch;pub use diverse::DiverseBeam;pub use diverse::DiverseBeamConfig;pub use length_penalty::LengthPenalty;pub use length_penalty::LengthPenaltyConfig;
Modules§
- beam
- Generic beam search with length-normalisation and diversity penalty.
- diverse
- length_
penalty - GNMT-style length penalty and coverage penalty for beam search scoring.