Write `n_batches * batch_size` on the same line.