zsl

> Hi @vr25, > I had the same problem and my solution is: > > 1. Check the version of `pytorch` with `cuda` compatibility > > * To check your...

> Hi, unfortunately I didnt have time to maintain the master branch of the repo. There is a branch where I implemented beam search: https://github.com/as-ideas/headliner/blob/improve_bert_beam/headliner/model/summarizer_bert.py, you can use summarizer.predict_beam_search() and...