4cdf99d4f8a25639ccf89a933a5cfc6553d657ad,onmt/translate/Beam.py,GNMTGlobalScorer,update_score,#GNMTGlobalScorer#Any#Any#,182

Before Change

        if "prev_penalty" in beam.global_state.keys():
            beam.scores.add_(beam.global_state["prev_penalty"])
            cov = beam.global_state["coverage"] + attn
            penalty = self.beta * (torch.max(cov, cov.clone().fill_(1.0)).sum(1) - cov.size(1))
            beam.scores.sub_(penalty)

    def update_global_state(self, beam):
        "Keeps the coverage vector as sum of attens"

After Change

        if "prev_penalty" in beam.global_state.keys():
            beam.scores.add_(beam.global_state["prev_penalty"])
            cov = beam.global_state["coverage"] + attn
            penalty = self.cov_penalty(beam,
                                       logprobs,
                                       beam.global_state["coverage"],
                                       self.beta)
            beam.global_state["prev_penalty"] = penalty
            beam.scores.sub_(penalty)

    def update_global_state(self, beam):

Instances