966436403e5e927e3e7d5b389dad6ef06aaa7e03,tests/test_sequence_generator.py,TestSequeneceGenerator,test_without_normalization,#TestSequeneceGenerator#,210

Before Change

        generator = SequenceGenerator(
            [self.model], self.tgt_dict, beam_size=2, normalize_scores=False
        )
        hypos = generator.forward(self.sample)
        eos, w1, w2 = self.tgt_dict.eos(), self.w1, self.w2
        // sentence 1, beam 1
        self.assertHypoTokens(hypos[0][0], [w1, eos])
        self.assertHypoScore(hypos[0][0], [0.9, 1.0], normalized=False)

After Change

    def test_without_normalization(self):
        // Sentence 1: unchanged from the normalized case
        // Sentence 2: beams swap order
        generator = SequenceGenerator(self.tgt_dict, beam_size=2, normalize_scores=False)
        hypos = generator.generate([self.model], self.sample)
        eos, w1, w2 = self.tgt_dict.eos(), self.w1, self.w2
        // sentence 1, beam 1
        self.assertHypoTokens(hypos[0][0], [w1, eos])

Instances