Skip to content

GitLab

"docs/git@developer.sourcefind.cn:change/sglang.git" did not exist on "75d171a9c59bb40952e89c3751c3a1c13c1d232d"
Commit 2fd3dcf3 authored by annarev's avatar annarev Committed by GitHub
Browse files

Merge pull request #966 from tensorflow/update-cifar10 

Update tf.concat_v2 to tf.concat
parents d43335c5 320545f0
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tutorials/image/cifar10/cifar10_multi_gpu_train.py
View file @ 2fd3dcf3
...@@ -124,7 +124,7 @@ def average_gradients(tower_grads): ...@@ -124,7 +124,7 @@ def average_gradients(tower_grads):
grads.append(expanded_g) grads.append(expanded_g)
# Average over the 'tower' dimension. # Average over the 'tower' dimension.
grad = tf.concat_v2(grads, 0) grad = tf.concat(grads, 0)
grad = tf.reduce_mean(grad, 0) grad = tf.reduce_mean(grad, 0)
# Keep in mind that the Variables are redundant because they are shared # Keep in mind that the Variables are redundant because they are shared
......
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