Fix indent

97f97d99 · minoring · 9691ef7a · 97f97d99
Commit 97f97d99 authored Oct 17, 2019 by minoring
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

official/mnist/mnist.py official/mnist/mnist.py +2 -1

No files found.
--- a/official/mnist/mnist.py
+++ b/official/mnist/mnist.py
@@ -144,7 +144,8 @@ def model_fn(features, labels, mode, params):
    return tf.estimator.EstimatorSpec(
        mode=tf.estimator.ModeKeys.TRAIN,
        loss=loss,
-        train_op=optimizer.minimize(loss,
+        train_op=optimizer.minimize(
+            loss,
            tf.compat.v1.train.get_or_create_global_step()))
  if mode == tf.estimator.ModeKeys.EVAL:
    logits = model(image, training=False)