PyTorch Beam Search Decoding Save
PyTorch implementation of beam search decoding for seq2seq models
Project README
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.
Open Source Agenda is not affiliated with "PyTorch Beam Search Decoding" Project. README Source: budzianowski/PyTorch-Beam-Search-Decoding
Stars
327
Open Issues
7
Last Commit
1 year ago
Repository
License
Open Source Agenda Badge
