Thanks, I've made the suggested changes. The process is no longer hanging, but I see the following error. (I haven't made any other code changes)

santhoshkolloju / Abstractive-Summarization-With-Transfer-Learning

Abstractive summarisation using Bert as encoder and Transformer Decoder

406 stars 98 forks source link

Open angeluau opened 5 years ago

santhoshkolloju commented 5 years ago

you should replace references = utils.list_strip_eos(references[0], eos_token_id) with

references = utils.list_strip_eos(references, eos_token_id)