never mind, we use `batch_size` in the other data layers so this is right, sorry