accelerate

Fix batch_size sanity check logic for `split_batches `

#2344

Merged