Skip to content

GitLab

Commit b528ede6 authored by Appledore's avatar Appledore Committed by GitHub
Browse files

Merge pull request #1998 from mari-linhares/patch-2 

Removing duplicated tower_losses.append(tower_loss) from cifar10_estimator
parents 2a5f2a95 1b221a2c
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 1 deletion
tutorials/image/cifar10_estimator/cifar10_main.py
View file @ b528ede6
...@@ -282,7 +282,6 @@ def _tower_fn(is_training, weight_decay, feature, label, tower_losses, ...@@ -282,7 +282,6 @@ def _tower_fn(is_training, weight_decay, feature, label, tower_losses,
tower_loss = tf.losses.sparse_softmax_cross_entropy( tower_loss = tf.losses.sparse_softmax_cross_entropy(
logits=logits, labels=label) logits=logits, labels=label)
tower_loss = tf.reduce_mean(tower_loss) tower_loss = tf.reduce_mean(tower_loss)
tower_losses.append(tower_loss)
model_params = tf.trainable_variables() model_params = tf.trainable_variables()
tower_loss += weight_decay * tf.add_n( tower_loss += weight_decay * tf.add_n(
......
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