07eed27d9fa6f1feaf2d09ba25ff26a643e70b3a,fairseq/search.py,DiverseBeamSearch,step,#DiverseBeamSearch#Any#Any#Any#,130

Before Change

            )

        // initialize diversity penalty
        if self.diversity_buf is None:
            self.diversity_buf = lprobs.new()
        self.diversity_buf.resize_(lprobs[:, 0, :].size())
        torch.zeros(lprobs[:, 0, :].size(), out=self.diversity_buf)

        scores_G, indices_G, beams_G = [], [], []

After Change

            )

        // initialize diversity penalty
        diversity_buf = torch.zeros(lprobs[:, 0, :].size()).to(lprobs)

        scores_G, indices_G, beams_G = [], [], []
        for g in range(self.num_groups):
            lprobs_g = lprobs[:, g :: self.num_groups, :]

Instances