Store batch_size in configuration