Commit a7c225c8 authored by Yong Tang's avatar Yong Tang
Browse files

Remove warnings in tf.losses.softmax_cross_entropy



This fix tries to address the issue raised in 16534 where
tf.losses.softmax_cross_entropy causes warnings due
to the calling of tf.nn.softmax_cross_entropy_with_logits.

This fix switches to tf.nn.softmax_cross_entropy_with_logits_v2
to remove the warning.

This fix fixes 16534.

Signed-off-by: default avatarYong Tang <yong.tang.github@outlook.com>
parent 8aa14cd6
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment