def METHOD_NAME(engine, batch):
    idx = (engine.state.iteration - 1) * batch_size
    y_true_batch = np_y[idx : idx + batch_size]
    y_pred_batch = np_y_pred[idx : idx + batch_size]
    return torch.from_numpy(y_pred_batch), torch.from_numpy(y_true_batch)