leviswind / pytorch-transformer

pytorch implementation of Attention is all you need
239 stars 58 forks source link

How to add some functionalities to this code? #5

Open liperrino opened 5 years ago

liperrino commented 5 years ago

Hi. Please i would like to add some features to this code that i have read on beam search. Like coverage penalty and length normalization. But i don't know where to start. Can you help please?