def num_examples(self):
    if (self.fold == 'train'):
        return 133125
    elif (self.fold == 'valid'):
        return 16759
    elif (self.fold == 'test'):
        return 16611
def num_examples(self):
    if (self.fold == 'train'):
        return 133125
    elif (self.fold == 'valid'):
        return 16759
    elif (self.fold == 'test'):
        return 16611