PyTorch Beam Search Decoding

Pesquisa PyTorch-Beam

Implementação PyTorch de decodificação de pesquisa de feixe para modelos seq2seq baseados em https://github.com/shawnwun/NNDIAL. A decodificação ocorre separadamente para cada frase e armazena os nós na fila priorizada.

Uso: você pode especificar recompensa adicional pela decodificação por meio de BeamSearchNode.eval. Funciona para modelo com e sem atenção.