beam search generating more that just the top sentence

lucidrains / reformer-pytorch

Reformer, the efficient Transformer, in Pytorch

MIT License

2.1k stars 254 forks source link

beam search generating more that just the top sentence #102

Closed anisdismail closed 4 years ago

anisdismail commented 4 years ago

In the generate function implemented here, I can't seem to find any way to have access to more the top sentence generated (the top 5 sentences for example). Are there any plans for implementing something like so, or it already exists?

lucidrains commented 4 years ago

@anisdismail I don't have a beam search yet, you should submit a PR :)

anisdismail commented 4 years ago

sure! will do very soon! Will this mean you are currently working on one or can I collaborate and try to work on one? (I am sorry I am new to open source projects...)

lucidrains commented 4 years ago

I have my hands full at the moment, you should go ahead and make one! maybe you could even release it as your own package