(1)加载数据集

如果你是AI零基础，请关注本专栏，将带你一起飞。

在PyTorch中，通常使用torchvision和torch.utils.data模块实现数据集的加载功能。在此模块中提供了用于加载和预处理常见数据集的工具，同时也支持自定义数据集的加载。

2.1.1  PyTorch加载数据集

在PyTorch程序中，模块torchvision.datasets提供了许多常见的预定义数据集，并提供了简单的API来加载这些数据集。以下是一些常用的数据集加载函数：

1. torchvision.datasets.ImageFolder：用于加载图像文件夹数据集，其中每个子文件夹表示一个类别，文件夹中的图像属于该类别。
2. torchvision.datasets.CIFAR10和torchvision.datasets.CIFAR100：用于加载CIFAR-10和CIFAR-100数据集，这是两个广泛使用的图像分类数据集。
3. torchvision.datasets.MNIST：用于加载MNIST手写数字数据集，其中包含了大量的手写数字图像及其对应的标签。
4. torchvision