why do we need to change the `batch_size`?