def METHOD_NAME(engine, batch):
    y_true_batch = next(y_true_batch_values)
    y_pred_batch = next(y_pred_batch_values)
    if output_transform is not None:
        return {"y_pred": torch.from_numpy(y_pred_batch), "y": torch.from_numpy(y_true_batch)}
    return torch.from_numpy(y_pred_batch), torch.from_numpy(y_true_batch)