open-compass / open-compass/opencompass
不同数据集使用不同的batch_size大小加速推理[Feature]
Open
@tonysy is already working on this.
Since Jul 21, 2024.
- Dominant language
- Python
- Stars
- 7.5k
- Forks
- 869
- Avg merge
- 17h 52m
- Merged PRs (30d)
- 13
Description
Describe the feature
目前评测多个数据集时，如果不使用vllm，只能在模型测添加batch_size，但是有的数据集较长，有的较短，同样的batch_size可能会利用gpu不充分，如何针对数据集设置batch_size。
Will you implement it?
- I would like to implement this feature and create a PR!
Contributor guide
No contributing guide indexed for this repository
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Assessment
This issue has not been assessed yet.