Predict beam size on Seq2SeqTrainer
Hi,There is any way to select the beam size of the evaluation step of Seq2SeqTrainer?
2021-07-15T17:03:21Z
You can set it intrainer.evaluate(num_beams=...)or set the default you like when instantiating your model:model = AutoModelForXXX(checkpoint, num_beams=...)