1、关于数据集
可在pytorch官网找到自己需要的
https://pytorch.org/
数据集查找方式
下载数据集
import torchvision
#使用CIFAR10数据集
#1、下载训练集,root:保存地址
train_set = torchvision.datasets.CIFAR10(root="E:/PycharmProjects/Pytoch_learning/dataset/CIFAR10", train=True,download=True)
#2、下载测试集
test_set = torchvision.datasets.CIFAR10(root="E:/PycharmProjects/Pytoch_learning/dataset/CIFAR10"