Skip to content

GitLab

Commit 97f97d99 authored by minoring's avatar minoring
Browse files

Fix indent

parent 9691ef7a
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
official/mnist/mnist.py
View file @ 97f97d99
...@@ -144,7 +144,8 @@ def model_fn(features, labels, mode, params): ...@@ -144,7 +144,8 @@ def model_fn(features, labels, mode, params):
return tf.estimator.EstimatorSpec( return tf.estimator.EstimatorSpec(
mode=tf.estimator.ModeKeys.TRAIN, mode=tf.estimator.ModeKeys.TRAIN,
loss=loss, loss=loss,
train_op=optimizer.minimize(loss, train_op=optimizer.minimize(
loss,
tf.compat.v1.train.get_or_create_global_step())) tf.compat.v1.train.get_or_create_global_step()))
if mode == tf.estimator.ModeKeys.EVAL: if mode == tf.estimator.ModeKeys.EVAL:
logits = model(image, training=False) logits = model(image, training=False)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment