Explorar o código

The call to this function has changed (#142)

tf.reduce_sum(tf.nn.softmax_cross_entropy_with_logits(logits=logits, labels=y_))
Luanrly %!s(int64=8) %!d(string=hai) anos
pai
achega
534b4febcd
Modificáronse 1 ficheiros con 1 adicións e 1 borrados
  1. 1 1
      examples/4_Utils/tensorboard_advanced.py

+ 1 - 1
examples/4_Utils/tensorboard_advanced.py

@@ -71,7 +71,7 @@ with tf.name_scope('Model'):
 
 with tf.name_scope('Loss'):
     # Softmax Cross entropy (cost function)
-    loss = tf.reduce_mean(tf.nn.softmax_cross_entropy_with_logits(pred, y))
+    loss = tf.reduce_sum(tf.nn.softmax_cross_entropy_with_logits(logits=pred, labels=y))
 
 with tf.name_scope('SGD'):
     # Gradient Descent