e8f4839bbf04d8f0929138e2494b95b31b0c58f3,ludwig/models/modules/loss_modules.py,SequenceLoss,call,#SequenceLoss#Any#Any#,137

Before Change

                dtype=y_pred.dtype)
            y_pred = tf.concat([y_pred, pad], axis=1)
            // obtain mask from y_true
            mask = tf.cast(tf.greater(y_true, 0), tf.float32)
        elif y_pred.shape[1] > y_true.shape[1]:
            pad = tf.zeros(
                [

After Change

        // add one to sequence length to account for EOS token
        mask = tf.cast(
            tf.sequence_mask(sequence_length_2D(y_true) + 1, y_true.shape[1]),
            dtype=tf.float32
        )

Instances