balanced sampler issue

[JaejunHwang]

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.