Browse Source

Update tf.concat_v2 to tf.concat

Neal Wu 8 năm trước cách đây
mục cha
commit
320545f014
1 tập tin đã thay đổi với 1 bổ sung1 xóa
  1. 1 1
      tutorials/image/cifar10/cifar10_multi_gpu_train.py

+ 1 - 1
tutorials/image/cifar10/cifar10_multi_gpu_train.py

@@ -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