瀏覽代碼

Update tf.concat_v2 to tf.concat

Neal Wu 8 年之前
父節點
當前提交
320545f014
共有 1 個文件被更改,包括 1 次插入1 次删除
  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