a584d6f73cfafa81c9e872ae6e9beb2feff32574,OpenNMT/onmt/Dataset.py,Dataset,__getitem__,#Dataset#Any#,23

Before Change

    def __getitem__(self, index):
        srcBatch = self._batchify(
            self.src[index*self.batchSize:(index+1)*self.batchSize]).t().contiguous()
        tgtBatch = self._batchify(
            self.tgt[index*self.batchSize:(index+1)*self.batchSize]).t().contiguous()

        return srcBatch, tgtBatch

    def __len__(self):

After Change

        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

Instances