def METHOD_NAME(B=256, S=0, label="", **kwargs):
    tic = time.perf_counter()
    res = []
    particles, _, _ = data[S : S + B]
    for chunk in particles.split(GPU_BATCH):
        res.append(do_pose_search(chunk, **kwargs))
    delta = time.perf_counter() - tic
    batch_rot = pose_rot[S : S + B]
    batch_trans = pose_trans[S : S + B]
    rot_hat, trans_hat, _ = [torch.cat(x) for x in zip(*res)]
    # print(rot_hat)
    # print(batch_rot)
    # print(pose_trans[:10])
    # print(trans_hat[:10])
    print(
        f"{label:20s}| "
        f"Rot MedSE= {medse(rot_hat, batch_rot):.5f} "
        f"Rot MSE= {mse(rot_hat, batch_rot):.5f} "
        f"Trans MedSE= {medse(trans_hat, batch_trans):.5f} "
        f"Trans MSE= {mse(trans_hat, batch_trans):.5f} "
        f"Trans offset= {trans_offset(trans_hat, batch_trans)} "
        f"time= {delta:.2f} s"
    )