Skip to content

Commit

Permalink
Merge pull request tensorflow#966 from tensorflow/update-cifar10
Browse files Browse the repository at this point in the history
Update tf.concat_v2 to tf.concat
  • Loading branch information
annarev authored Jan 31, 2017
2 parents d43335c + 320545f commit 2fd3dcf
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tutorials/image/cifar10/cifar10_multi_gpu_train.py
Original file line number Diff line number Diff line change
Expand Up @@ -124,7 +124,7 @@ def average_gradients(tower_grads):
grads.append(expanded_g)

# Average over the 'tower' dimension.
grad = tf.concat_v2(grads, 0)
grad = tf.concat(grads, 0)
grad = tf.reduce_mean(grad, 0)

# Keep in mind that the Variables are redundant because they are shared
Expand Down

0 comments on commit 2fd3dcf

Please sign in to comment.