Beam search seems not process the <end> token?

Hi,

Thank you for your code! It seems the function of beam search does not end the sentence when it meets the <end> token, causing the sentences by beam search (e.g. BS=5) not to be different from each other. Could you check it again? And by the way, could you add DBS in the reference stage?

Thank you