How to set the length of generated tokens?

OFA-Sys / OFA

Official repository of OFA (ICML 2022). Paper: OFA: Unifying Architectures, Tasks, and Modalities Through a Simple Sequence-to-Sequence Learning Framework

Apache License 2.0

2.39k stars 248 forks source link

How to set the length of generated tokens? #378

Open xcvil opened 1 year ago

xcvil commented 1 year ago

Dear coauthors,

In pretraining/finetuning stage, for vision-language task (especially for visual_grounding and caption), can I set the length of generated tokens? Because I want a longer generated sentence.
Also for captioning evaluation, how can I set the output length?
I found there are some arguments like max-length-a and max-length-b. Could you help me understand the meaning?

Thanks a lot!

logicwong commented 1 year ago

@xcvil Of course you can control the generation length at any stage. For the meaning of max-length-a, max-length-b, and others. You can refer to here