change the batch size