Crash while using mention table

[xu1998hz]

I used "mention2wikidataID_with_titles_label_alias_redirect.pkl" and "wikidataID2lang_title-normalized_with_redirect.pkl" to add constraints on the search space. However, I received the following errors on one example.

The example was "LONDON, England (CNN) -- There are some circuits that are cornerstones of the Formula 1 calendar: the Monacos and [START] Silverstones [END] whose every twist and turn are known intimately by the drivers and engineers."

File ~/GENRE/fairseq/fairseq/sequence_generator.py:902, in (.0) 899 else: 900 print(finalized) 901 attn = [ --> 902 finalized[i // beam_size][i % beam_size]["attention"].transpose(1, 0) 903 for i in range(bsz * beam_size) 904 ] 906 if src_tokens.device != "cpu": 907 src_tokens = src_tokens.to("cpu")

IndexError: list index out of range

[nicola-decao]

Your installation of fairseq seems odd. Why do you have print(finalized)?

[nicola-decao]

Did you use my version of fariseq?

[xu1998hz]

I just printed myself to figure out some potential bugs

[xu1998hz]

Yes, version is correct. The code works OK without using mention table

[nicola-decao]

Show me the code you are running