pytorch_beam_search
pytorch_beam_search copied to clipboard
jarobyte91
A simple library that implements search algorithms for sequence models written in PyTorch.
The setup file pins the version of pytorch to >=1.8.1. I don't think any of the APIs used require a very specific pytorch release (would 2.0.1 not work equally well?)....
Hi, great work on the beam search implementation. I was going through your code to understand your implementation and noticed that you a) Don't terminate a beam when the EOS...
Metadata
Owner
Metadata
A simple library that implements search algorithms for sequence models written in PyTorch.