58cb18761fc72e5f9b8dd92a222a14b0240d8436,onmt/Translator.py,Translator,translate,#Translator#Any#Any#,246

Before Change

        //  (2) translate
        pred, predScore, attn, goldScore = self.translateBatch(batch)
        pred, predScore, attn, goldScore = list(zip(
            *sorted(zip(pred, predScore, attn, goldScore, batch.indices),
                    key=lambda x: x[-1])))[:-1]

        //  (3) convert indexes to words
        predBatch = []
        for b in range(batchSize):

After Change

        //  (3) convert indexes to words
        predBatch = []
        src = batch.src[0].data.index_select(1, perm)
        for b in range(batchSize):
            src_vocab = data.src_vocabs[inds[b]]
            predBatch.append(
                [self.buildTargetTokens(pred[b][n], src[:, b],

Instances