pytorch实现mnist数据集分类
1.数据准备
pytorch框架内置的torchvision中的datasets类中有一些常见的数据集。本文采用随机梯度下降的算法来训练数据,数据集构造如下所示:
# 将数据集转换成张量并且归一化
transform = transforms.Compose({
transforms.ToTensor(),
transforms.Normalize((0.1307,), (0.3081,))}
)
# 下载数据集,download= True表示从网络下载,本文已经下载好了这
原创
2022-03-20 12:20:35 ·
2511 阅读 ·
2 评论