数据集下载

1、直接下载pytorch官方现成的数据集:

#数据集下载,训练集
trainset = torchvision.datasets.name(root='./data',train=True,download=True,transform=transformer)
#数据集导入
trainloader = torch.utils.data.DataLoader(trainset,batch_size=36,shuffle=True,num_workers=0)
#测试集下载
tsetset = torchvision.datasets.name(root='./data',train=False,download=True,transform=transformer)
#测试集导入
trainloader = torch.utils.data.DataLoader(testset,batch_size=10,shuffle=False,num_workers=0)

下载到了同目录下的data文件夹下。
其中,输入到torchvision.datasets.时,会出现选择数据集的选项,选择自己所需的数据集即可(如下图所示),下载完之后,把download更改为False,download=False。
数据集下载_第1张图片

2、提前下载数据集,导入路径。

你可能感兴趣的:(pytorch,深度学习,transformer,python)