8055210d965130ded71d06ea4cc59c2e82e65b15,models/experimental/mask_rcnn/mask_rcnn_model.py,,_model_fn,#Any#Any#Any#Any#Any#,315

Before Change

    tpu_spec: the TPUEstimatorSpec to run training, evaluation, or prediction.
  
  if mode == tf.estimator.ModeKeys.TRAIN:
    if "features" not in features:
      raise ValueError(""features" is missing in TRAIN input.")
    if "labels" not in features:
      raise ValueError(""labels" is missing in TRAIN input.")
    labels = features["labels"]
    features = features["features"]
  else:
    if params["include_groundtruth_in_features"] and ("labels" in features):
      // In include groundtruth for eval.
      labels = features["labels"]

After Change

  else:
    model_outputs = build_model_graph(
        features, labels, mode == tf.estimator.ModeKeys.TRAIN, params)
    model_outputs.update({
        "source_id": features["source_ids"],
        "image_info": features["image_info"],
    })

  // First check if it is in PREDICT mode.
  if mode == tf.estimator.ModeKeys.PREDICT:
    predictions = {}

Instances