examples/research_projects/codeparrot/scripts/bpe_training.py/batch_iterator
def batch_iterator(batch_size=10):
    for _ in tqdm(range(0, args.n_examples, batch_size)):
        yield [next(iter_dataset)[args.text_column] for _ in range(batch_size)]