PyTorch Beam Search Decoding

Recherche PyTorch-Beam

Implémentation PyTorch du décodage de recherche de faisceau pour les modèles seq2seq basés sur https://github.com/shawnwun/NNDIAL. Le décodage s'effectue séparément pour chaque phrase et stocke les nœuds dans une file d'attente prioritaire.

Utilisation : vous pouvez spécifier une récompense supplémentaire pour le décodage via BeamSearchNode.eval. Fonctionne pour modèle avec et sans attention.