/pytorch_beam_search

A lightweight implementation of Beam Search for sequence models in PyTorch.

Primary LanguagePythonMIT LicenseMIT

Watchers