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

Doesn't have to be perfect.