diff options
| author |  Jianwei Xie <xiejw@google.com> | 2017-10-18 13:07:01 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-10-18 13:10:59 -0700 |
| commit | f6968a25c9dfc962851806d094dc98f5b502a4f9 (patch) | |
| tree | 1f6d779b51ae53e286668f25c7fbb54ce46224d9 /tensorflow/examples/learn | |
| parent | b7e85339b286f34f215cca3dcb700dbd8f276de3 (diff) | |
Add logging verbosity to mnist.py
PiperOrigin-RevId: 172643922
Diffstat (limited to 'tensorflow/examples/learn')
| -rw-r--r-- | tensorflow/examples/learn/mnist.py | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow/examples/learn/mnist.py b/tensorflow/examples/learn/mnist.py index 5344526b52..88425ea0d0 100644 --- a/tensorflow/examples/learn/mnist.py +++ b/tensorflow/examples/learn/mnist.py @@ -97,6 +97,8 @@ def conv_model(features, labels, mode): def main(unused_args): + tf.logging.set_verbosity(tf.logging.INFO) + ### Download and load MNIST dataset. mnist = tf.contrib.learn.datasets.DATASETS['mnist']('/tmp/mnist') train_input_fn = tf.estimator.inputs.numpy_input_fn( @@ -115,6 +117,7 @@ def main(unused_args): feature_columns = [ tf.feature_column.numeric_column( X_FEATURE, shape=mnist.train.images.shape[1:])] + classifier = tf.estimator.LinearClassifier( feature_columns=feature_columns, n_classes=N_DIGITS) classifier.train(input_fn=train_input_fn, steps=200) |