diff options
Diffstat (limited to 'tensorflow/examples/learn/resnet.py')
| -rwxr-xr-x | tensorflow/examples/learn/resnet.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/examples/learn/resnet.py b/tensorflow/examples/learn/resnet.py index 49d89ef660..f822903208 100755 --- a/tensorflow/examples/learn/resnet.py +++ b/tensorflow/examples/learn/resnet.py @@ -144,7 +144,7 @@ def res_net(x, y, activation=tf.nn.relu): target = tf.one_hot(y, depth=10, dtype=tf.float32) logits = tf.contrib.layers.fully_connected(net, 10, activation_fn=None) - loss = tf.contrib.losses.softmax_cross_entropy(logits, target) + loss = tf.losses.softmax_cross_entropy(target, logits) return tf.softmax(logits), loss |