def num_episodes(self):
    if (self.fold == 'train'):
        return 66478
    elif (self.fold == 'valid'):
        return 8310
    elif (self.fold == 'test'):
        return 8309
def num_episodes(self):
    if (self.fold == 'train'):
        return 66478
    elif (self.fold == 'valid'):
        return 8310
    elif (self.fold == 'test'):
        return 8309