How do you modify the Batch_size with the xmpl-x demo? I configured it in the .yaml but it doesn't seem to work. What needs to be modified?