diff options
| author |  Vijay Vasudevan <vrv@google.com> | 2017-02-17 17:05:49 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-02-17 17:23:48 -0800 |
| commit | 93a975e114ee1c35f01ed3bdd47170e6f7129014 (patch) | |
| tree | e34255aff698fe6a4a586e7940337fd278947f58 /tensorflow/examples/learn/mnist.py | |
| parent | eb9624017a0040e805fda622a5f9ec6681e24246 (diff) | |
Merge changes from github.
Change: 147897309
Diffstat (limited to 'tensorflow/examples/learn/mnist.py')
| -rw-r--r-- | tensorflow/examples/learn/mnist.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/examples/learn/mnist.py b/tensorflow/examples/learn/mnist.py index 4b3f1835e2..6e5fe7891b 100644 --- a/tensorflow/examples/learn/mnist.py +++ b/tensorflow/examples/learn/mnist.py @@ -67,7 +67,7 @@ def conv_model(feature, target, mode): # Compute logits (1 per class) and compute loss. logits = layers.fully_connected(h_fc1, 10, activation_fn=None) - loss = tf.contrib.losses.softmax_cross_entropy(logits, target) + loss = tf.losses.softmax_cross_entropy(target, logits) # Create a tensor for training op. train_op = layers.optimize_loss( |