基于Pyrorch深度学习之分类问题一（简单套路）

本文基于Pytorch构建简单的分类模型，详细阐述了分类问题所需要的步骤，以及注意事项。完整代码请移步Go-for-i。

---

步骤开始：

1.准备需要分类的数据集，其中包括训练数据集和测试数据集。

# 下面是训练集
training_data = datasets.FashionMNIST(
    root="data",
    train=True,
    download=True,
    transform=ToTensor(),
)

# 下面是测试集，同样需要下载
test_data = datasets.FashionMNIST(
    root="data",
    train=False,
    download=True,
    transform=ToTensor(),
)

2.创建数据加载器，将上述数据集按照一定的批次扔进模型中。其中，batch_size表示每次扔进模型中的数量。

batch_size = 64
# 给训练集和测试集分别创建一个数据集加载器
train_dataloader = DataLoader(training_data, batch_size=batch_size)
test_dataloader = Da