Predict beam size on Seq2SeqTrainer

🤗Transformers
1

Hi,

There is any way to select the beam size of the evaluation step of Seq2SeqTrainer?

2

You can set it in trainer.evaluate(num_beams=...) or set the default you like when instantiating your model:

model = AutoModelForXXX(checkpoint, num_beams=...)
1 Like