PyTorch Beam Search Decoding

Búsqueda de haz de PyTorch

Implementación de PyTorch de decodificación de búsqueda de haz para modelos seq2seq basado en https://github.com/shawnwun/NNDIAL. La decodificación se realiza por separado para cada oración y almacena los nodos en una cola priorizada.

Uso: Puede especificar una recompensa adicional por decodificar a través de BeamSearchNode.eval. Funciona para modelo con y sin atención.