in weak_to_strong/eval.py [0:0]
def to_batch(x, batch_size): for i in range(0, len(x), batch_size): yield x[i : i + batch_size]