added output_dir

sourcedexter · May 13, 2017 · 730e01a · 730e01a
1 parent 188ce45
commit 730e01a
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 1 deletion.
diff --git a/output_dir/all_ouput_stored_here b/output_dir/all_ouput_stored_here
diff --git a/retrain.py b/retrain.py
@@ -664,7 +664,7 @@ def add_final_training_ops(class_count, final_tensor_name, bottleneck_tensor):
 
   with tf.name_scope('cross_entropy'):
     cross_entropy = tf.nn.softmax_cross_entropy_with_logits(
-      logits, ground_truth_input)
+      logits=logits, labels=ground_truth_input)
     with tf.name_scope('total'):
       cross_entropy_mean = tf.reduce_mean(cross_entropy)
   tf.summary.scalar('cross_entropy', cross_entropy_mean)