def cifar10(tnum=2): dataset = mt_dataset.cifar10() channel_stats = dict(mean=[0.4914, 0.4822, 0.4465], std=[0.2470, 0.2435, 0.2616]) dataset['train_transformation'] = data.TransformNTimes(transforms.Compose([ data.RandomTranslateWithReflect(4), transforms.RandomHorizontalFlip(), transforms.ToTensor(), transforms.Normalize(**channel_stats) ]), n=tnum) dataset['datadir'] = 'third_party/' + dataset['datadir'] return dataset
def mnist(tnum=2): channel_stats = dict(mean=[0.5, 0.5, 0.5], std=[0.5, 0.5, 0.5]) train_transformation = data.TransformNTimes(transforms.Compose( [transforms.ToTensor(), transforms.Normalize(**channel_stats)]), n=tnum) eval_transformation = transforms.Compose( [transforms.ToTensor(), transforms.Normalize(**channel_stats)]) return { 'train_transformation': train_transformation, 'eval_transformation': eval_transformation, 'datadir': 'third_party/data-local/images/mnist', 'num_classes': 10, }