search

yaserkl / RLSeq2Seq

Deep Reinforcement Learning For Sequence to Sequence Models
https://arxiv.org/abs/1805.09461
MIT License
767 stars 160 forks source link

beam search decoder #1

Closed lsq357 closed 6 years ago

lsq357 commented 6 years ago

Thanks for your excellent work ! Are you planning to support beam search decoder when test stage?

yaserkl commented 6 years ago

Thanks, all models support beam search decoding. Please check the beam_search.py and decoder.py files.