# Predict beam size on Seq2SeqTrainer

**URL:** </static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Ft%2Fpredict-beam-size-on-seq2seqtrainer%2F8393%3E%5C
**Category:** 🤗Transformers\
**Created:** [July 15, 2021, 4:17pm UTC](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Ft%2Fpredict-beam-size-on-seq2seqtrainer%2F8393 "2021-07-15T16:17:22Z")\
**Posts on this page:** 2\
**Page:** 1

<div class="post-metadata">

**Author:** ![SSamDav](https://sea2.discourse-cdn.com/hellohellohello/user_avatar/discuss.huggingface.co/ssamdav/32/29888_2.png) [@SSamDav](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Fu%2FSSamDav)%5C
**Post date:** [July 15, 2021, 4:17pm UTC](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Ft%2Fpredict-beam-size-on-seq2seqtrainer%2F8393%2F1 "2021-07-15T16:17:22Z")

</div>

Hi,

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

---

<div class="post-metadata">

**Author:** ![sgugger](https://sea2.discourse-cdn.com/hellohellohello/user_avatar/discuss.huggingface.co/sgugger/32/2291_2.png) [@sgugger](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Fu%2Fsgugger)%5C
**Post date:** [July 15, 2021, 5:03pm UTC](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Ft%2Fpredict-beam-size-on-seq2seqtrainer%2F8393%2F2 "2021-07-15T17:03:21Z")

</div>

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

```auto
model = AutoModelForXXX(checkpoint, num_beams=...)

```
