如何针对每个数据集设置不同的batch_size？

[JBoRu]

目前评测多个数据集时，只能在模型测添加batch_size，但是有的数据集较长，有的较短，同样的batch_size可能会利用gpu不充分，如何针对数据集设置batch_size。