def to_batch()

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]