dfe2a43fb378b557f3dfc4315417afb291cd16b7,official/mnist/mnist_tpu.py,,model_fn,#Any#Any#Any#Any#,81

Before Change

  model_fn constructs the ML model used to predict handwritten digits.

  del params
  if mode == tf.estimator.ModeKeys.PREDICT:
    raise RuntimeError("mode {} is not supported yet".format(mode))
  image = features
  if isinstance(image, dict):
    image = features["image"]

After Change

  if mode == tf.estimator.ModeKeys.PREDICT:
    logits = model(image, training=False)
    predictions = {
        "class_ids": tf.argmax(logits, axis=1),
        "probabilities": tf.nn.softmax(logits),
        "label": labels,
    }

Instances