10781ce4f0b85b5468ed8d8d485412d8cb307ae6,src/DataLoaderIAM.py,DataLoaderIAM,getIteratorInfo,#DataLoaderIAM#,117

Before Change

    def getIteratorInfo(self):
        "current batch index and overall number of batches"
        return (self.currIdx // self.batchSize + 1, len(self.samples) // self.batchSize)

    def hasNext(self):
        "iterator"
        return self.currIdx + self.batchSize <= len(self.samples)

After Change

    def getIteratorInfo(self):
        "current batch index and overall number of batches"
        if self.currSet == "train":
            numBatches = int(np.floor(len(self.samples) / self.batchSize))  // train set: only full-sized batches
        else:
            numBatches = int(np.ceil(len(self.samples) / self.batchSize))  // val set: allow last batch to be smaller
        currBatch = self.currIdx // self.batchSize + 1
        return currBatch, numBatches

    def hasNext(self):

Instances