这段代码是一个完整的 PyTorch 教程,从数据加载到模型训练、评估、保存和加载。以下是详细讲解:
1. 数据加载与处理
PyTorch 提供了两个数据处理原语:torch.utils.data.Dataset
和 torch.utils.data.DataLoader
。
- Dataset:存储样本及其对应的标签,例如
datasets.FashionMNIST
。
- DataLoader:为 Dataset 添加迭代器功能,支持批处理、采样、数据随机打乱和多线程数据加载。
代码关键点:
training_data = datasets.FashionMNIST(root="data"