I see, will try to find a suitable function to be the batch_size