max_num_batches