509453992838a524f6442d4e0f07a034390ae1f7,layers/eight_mile/tf/layers.py,CRF,neg_log_loss,#CRF#Any#Any#Any#,2274

Before Change

        :return: torch.FloatTensor: [B]
        
        fwd_score = self((unary, tf.cast(lengths, tf.int32)), training=True)
        gold_score = self.score_sentence(unary, tags, lengths)
        log_likelihood = gold_score - fwd_score
        return -tf.reduce_mean(log_likelihood)

After Change

        
        lengths = tf.cast(lengths, tf.int32)
        max_length = tf.reduce_max(lengths)
        fwd_score = self((unary, lengths), training=True)
        tags = tags[:, :max_length]
        gold_score = self.score_sentence(unary, tags, lengths)
        log_likelihood = gold_score - fwd_score

Instances