c751a1dd8a724388437916e9d29653c45fc4f14f,onmt/Dataset.py,Dataset,__getitem__,#Dataset#Any#,27

Before Change

        assert index < self.numBatches, "%d > %d" % (index, self.numBatches)
        srcBatch = self._batchify(
            self.src[index*self.batchSize:(index+1)*self.batchSize], align_right=True)
        tgtBatch = self._batchify(
            self.tgt[index*self.batchSize:(index+1)*self.batchSize])

        if self.cuda:
            srcBatch = srcBatch.cuda()
            tgtBatch = tgtBatch.cuda()

        // FIXME
        srcBatch = srcBatch.t().contiguous()
        tgtBatch = tgtBatch.t().contiguous()

        return srcBatch, tgtBatch

    def __len__(self):

After Change

        srcBatch = self._batchify(
            self.src[index*self.batchSize:(index+1)*self.batchSize], align_right=True)

        if self.tgt:
            tgtBatch = self._batchify(
                self.tgt[index*self.batchSize:(index+1)*self.batchSize])
        else:
            tgtBatch = None

        return srcBatch, tgtBatch

    def __len__(self):
        return self.numBatches

Instances