From 3c6419bd4053beb858b7ef911914e91496287389 Mon Sep 17 00:00:00 2001 From: Dhruv Date: Wed, 12 Sep 2018 13:55:05 -0700 Subject: [PATCH] Upgrade softmax_cross_entropy_with_logits_v2 + tf.nn.softmax_cross_entropy_with_logits_v2 --- scripts/retrain.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/scripts/retrain.py b/scripts/retrain.py index 5fa9b0f13..c7b450606 100644 --- a/scripts/retrain.py +++ b/scripts/retrain.py @@ -786,7 +786,7 @@ def add_final_training_ops(class_count, final_tensor_name, bottleneck_tensor, tf.summary.histogram('activations', final_tensor) with tf.name_scope('cross_entropy'): - cross_entropy = tf.nn.softmax_cross_entropy_with_logits( + cross_entropy = tf.nn.softmax_cross_entropy_with_logits_v2( labels=ground_truth_input, logits=logits) with tf.name_scope('total'): cross_entropy_mean = tf.reduce_mean(cross_entropy)