[go: up one dir, main page]

Skip to content

budzianowski/PyTorch-Beam-Search-Decoding

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 

Repository files navigation

PyTorch-Beam-Search

PyTorch implementation of beam search decoding for seq2seq models based on https://github.com/shawnwun/NNDIAL. Decoding goes seperately for each sentence and stores the nodes in prioritized queue.

Usage: You can specify additional reward for decoding through BeamSearchNode.eval. Works for model with and without attention.

Releases

No releases published

Packages

No packages published

Languages