def _update_peak_max_pool_kernel_size(model_config, kernel_size):
  """Updates the max pool kernel size (NMS) for keypoints in CenterNet."""
  meta_architecture = model_config.WhichOneof("model")
  if meta_architecture == "center_net":
    if len(model_config.center_net.keypoint_estimation_task) == 1: