'Args:
train_or_test (str): either \'train\' or \'test\'.
shuffle (bool): shuffle the dataset.'

def __init__(self, train_or_test, shuffle=True, dir=None):

    super(Cifar10, self).__init__(train_or_test, shuffle, dir, 10)
