Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
A minor mistake in cross entropy loss (#357)
tf.reduce_mean(-tf.reduce_sum(y_true * tf.math.log(y_pred),1)) or else it simply finds the sum and the reduced mean remains the sum itself.
- Loading branch information