0e7bfac76b97e7630102687f7e6309e00e190091,onmt/translate/translator.py,Translator,_score_target,#Translator#Any#Any#Any#Any#Any#,676

Before Change

                                      src_map=src_map)
        tgt_pad = self.fields["tgt"].vocab.stoi[inputters.PAD_WORD]
        for log_prob, tgt in zip(log_probs, batch.tgt[1:]):
            tgt = tgt.unsqueeze(1)
            scores = log_prob.gather(1, tgt)
            scores.masked_fill_(tgt.eq(tgt_pad), 0)
            gold_scores += scores.view(-1)

After Change

        log_probs[:, :, tgt_pad] = 0
        gold = batch.tgt[1:].unsqueeze(2)
        gold_scores = log_probs.gather(2, gold)
        gold_scores = gold_scores.sum(dim=0).view(-1)

        return gold_scores

Instances