Awesome Open Source
Awesome Open Source

PyTorch-Beam-Search

PyTorch implementation of beam search decoding for seq2seq models based on https://awesomeopensource.com/project/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.


Get A Weekly Email With Trending Projects For These Topics
No Spam. Unsubscribe easily at any time.
Python (1,142,144
Pytorch (11,083
Natural Language Processing (4,610
Search (2,695
Seq2seq (538
Torch (371
Decoding (339
Neural (181
Beam (163
Related Projects