Commit c1ceedf7 authored by Amir MOHAMMADI's avatar Amir MOHAMMADI
Browse files

loss is summarized by default in tensorflow 1.4

parent d32a1cce
Pipeline #13536 failed with stages
in 2 minutes and 12 seconds
......@@ -101,10 +101,6 @@ def model_fn(features, labels, mode, params=None, config=None):
loss = tf.losses.sparse_softmax_cross_entropy(
logits=logits, labels=labels)
with tf.name_scope('train_metrics'):
# Create a tensor named train_loss for logging purposes
tf.summary.scalar('train_loss', loss)
# Configure the Training Op (for TRAIN mode)
if mode == tf.estimator.ModeKeys.TRAIN:
optimizer = tf.train.GradientDescentOptimizer(
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment