When I try to use --balance-batch option, there is a problem.
In MultiFDataset
def _get_training_samples(self, index): indices = [] for class_ in self.labels_set: real_index = self.label_to_indices[class_][int(index * self.ratios[class_])] indices.append(real_index) return indices
at the part of (index * self.ratios[class_]), their range is over the self.label_to_indices itself.
When I try to use --balance-batch option, there is a problem.
In MultiFDataset
def _get_training_samples(self, index): indices = [] for class_ in self.labels_set: real_index = self.label_to_indices[class_][int(index * self.ratios[class_])] indices.append(real_index) return indicesat the part of (index * self.ratios[class_]), their range is over the self.label_to_indices itself.