From 207ed8cb78ceb4980e40c89f867b06202e660395 Mon Sep 17 00:00:00 2001 From: Sam Shleifer Date: Thu, 3 Sep 2020 12:42:09 -0400 Subject: [PATCH] [s2s] use --eval_beams command line arg (#6926) --- examples/seq2seq/finetune.py | 1 + 1 file changed, 1 insertion(+) diff --git a/examples/seq2seq/finetune.py b/examples/seq2seq/finetune.py index 16e407fa1e..8181b8e860 100644 --- a/examples/seq2seq/finetune.py +++ b/examples/seq2seq/finetune.py @@ -202,6 +202,7 @@ class SummarizationModule(BaseTransformer): attention_mask=batch["attention_mask"], use_cache=True, decoder_start_token_id=self.decoder_start_token_id, + num_beams=self.eval_beams, ) gen_time = (time.time() - t0) / batch["input_ids"].shape[0] preds: List[str] = self.ids_to_clean_text(generated_ids)