bigcode-project / bigcode-evaluation-harness

A framework for the evaluation of autoregressive code generation language models.
Apache License 2.0
771 stars 201 forks source link

Variable max_length_generation #49

Open Muennighoff opened 1 year ago

Muennighoff commented 1 year ago

Allow max_length_generation to change from batch to batch to speed-up tasks where length changes a lot. For tasks scored with exact match, we even know the maximum length for each sample, so it would be nice to just limited the max length for those samples. Need to be careful to make it work with batching.