Skip to content
This repository has been archived by the owner on Jun 10, 2021. It is now read-only.

Can shallow fusion support arbitrary language models other than RNNLM? #507

Open
getao opened this issue Jan 31, 2018 · 2 comments
Open

Comments

@getao
Copy link

getao commented Jan 31, 2018

Hello. I wonder if I can use arbitrary language models (e.g., an n-gram LM) to help decoding using the shallow fusion strategy. It seems that currently OpenNMT only supports shallow fusion with RNNLM trained by itself.

Thank you in advance.

@jsenellart
Copy link
Contributor

yes - as of today, we only support onmt rnnlm since code is integrated. through hooks mechanism it would not be hard to extend to other LM though. Do you have one in particular in mind?

@getao
Copy link
Author

getao commented Feb 2, 2018

Thanks @jsenellart . I'll have a look at the hook mechanism.

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Development

No branches or pull requests

2 participants