Track beam search in OpenNMT-py

Hi there,
Is it possible to trace the exploration of beam search in OpenNMT-py like in OpenNMT-lua?
If I want to print the probability of each word in the n-best predictions, which module I should to modify?

Thank you in advance for your attention :smiley:

if you’re interested, you can start with this: https://github.com/OpenNMT/OpenNMT-py/pull/1240
it has never been finished

OK, I will read it, with many thanks!