pytorch之mnist数据集存放位置以及本地加载

train_dataset = torchvision.datasets.MNIST(root='./data',
                                           train=True,
                                           transform=transforms.ToTensor(),
                                           download=True)

把mnist数据集连同文件夹一起放到./data下即可

你可能感兴趣的:(python)