In addition, I would like to add that my data set is torchvision.datasets.CIFAR10(args.data_path, train=True, transform=train_transform, download=True), cifar.python is only automatically downloaded. So, this should be the official data set in torchivision.