625451ab74f1942eb951337554d72005076a51b3,onmt/Translator.py,Translator,translateBatch,#Translator#Any#Any#,95

Before Change

        elif hasattr(batch, "src_img"):
            src = onmt.IO.make_features(batch, "src_img")
            src_lengths = None
        elif hasattr(batch, "src_audio"):
            src = onmt.IO.make_features(batch, "src_audio")
            src_lengths = None
        encStates, context = self.model.encoder(src, src_lengths)

After Change

                                        src, context, encStates)

        if src_lengths is None:
            src_lengths = torch.Tensor(batch_size).type_as(context.data)\
                                                  .long()\
                                                  .fill_(context.size(0))

        //  (1b) Initialize for the decoder.
        def var(a): return Variable(a, volatile=True)

Instances