6aa255e3d273a78612a68e648a761f3d830e319c,thumt/models/transformer.py,,model_graph,#Any#Any#Any#Any#,169

Before Change

    )

    ce = tf.reshape(ce, tf.shape(tgt_seq))
    loss = tf.reduce_sum(ce * tgt_mask) / tf.reduce_sum(tgt_mask)

    return loss

After Change

    )

    ce = tf.reshape(ce, tf.shape(tgt_seq))
    loss = get_loss(features, params, ce, tgt_mask)

    return loss

Instances