f390cc34b75c2e5dd9b9db1b66ec41525174ee61,onmt/Translator.py,Translator,_runTarget,#Translator#Any#Any#,59

Before Change

        tt = torch.cuda if self.opt.cuda else torch
        goldScores = tt.FloatTensor(batch.batch_size).fill_(0)
        decOut, decStates, attn = self.model.decoder(
            batch.tgt[:-1], src, context, decStates)

        // print(decOut.size(), batch.tgt[1:].data.size())
        // aeq(decOut.size(), batch.tgt[1:].data.size())

After Change

        _, src_lengths = batch.src
        src = onmt.IO.make_features(batch, "src")
        tgt_in = onmt.IO.make_features(batch, "tgt")[:-1]

        //  (1) run the encoder on the src
        encStates, context = self.model.encoder(src, src_lengths)
        decStates = self.model.init_decoder_state(context, encStates)

Instances