Skip to content
Snippets Groups Projects
Commit c1ceedf7 authored by Amir MOHAMMADI's avatar Amir MOHAMMADI
Browse files

loss is summarized by default in tensorflow 1.4

parent d32a1cce
No related branches found
No related tags found
1 merge request!28Take estimators from config in generic scripts
Pipeline #
......@@ -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(
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment