瀏覽代碼

old argument targets->labels of tf.nn.softmax_cross_entropy_with_logits for tf 1.0

stream 8 年之前
父節點
當前提交
ceed1a31db
共有 1 個文件被更改,包括 1 次插入1 次删除
  1. 1 1
      transformer/cluttered_mnist.py

+ 1 - 1
transformer/cluttered_mnist.py

@@ -123,7 +123,7 @@ y_logits = tf.matmul(h_fc1_drop, W_fc2) + b_fc2
 
 # %% Define loss/eval/training functions
 cross_entropy = tf.reduce_mean(
-    tf.nn.softmax_cross_entropy_with_logits(logits=y_logits, targets=y))
+    tf.nn.softmax_cross_entropy_with_logits(logits=y_logits, labels=y))
 opt = tf.train.AdamOptimizer()
 optimizer = opt.minimize(cross_entropy)
 grads = opt.compute_gradients(cross_entropy, [b_fc_loc2])