Fix beam size argument when translating with fairseq

ZurichNLP · May 2, 2024 · 908c67e · 908c67e
1 parent 83d5bed
commit 908c67e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/experiments/reference_aggregation/baseline_beam_search.py b/experiments/reference_aggregation/baseline_beam_search.py
@@ -18,7 +18,7 @@ def main(testset: str, language_pair: str, beam_size: int = 4, limit_segments: i
     translations_dir.mkdir(exist_ok=True)
     out_path = translations_dir / f"{dataset}.beam{beam_size}.{dataset.tgt_lang}"
 
-    translations = model.translate(dataset.source_sentences, beam_size=beam_size)
+    translations = model.translate(dataset.source_sentences, beam=beam_size)
     assert len(translations) == len(dataset.source_sentences)
 
     with open(out_path, "w") as f: