2019年2月7日
用Pytorch训练MNIST分类模型
摘要： 本次分类问题使用的数据集是MNIST，每个图像的大小为$28 28$。 编写代码的步骤如下 1. 载入数据集，分别为训练集和测试集 2. 让数据集可以迭代 3. 定义模型，定义损失函数，训练模型 代码 输出如下 阅读全文
posted @ 2019-02-07 15:59 MartinLwx 阅读(3429) 评论(0) 推荐(0) 编辑