基于PyTorch的手写数字识别
# 基于 PyTorch 的手写数字识别
# MNIST 数据集
MNIST 是一个经典的手写数字识别数据集,包含 60,000 张用于训练的图片和 10,000 张用于测试的图片。每张图片分辨率为 28×28 像素,内容是数字 0–9 的手写体。我们将基于该数据集对模型进行训练和验证。
# 神经网络的定义
# 输入
一张图片大小: 1 × 28 × 28 (灰度图,通道数 = 1)
# 第一层卷积
nn.Conv2d(in_channels=1, out_channels=16, kernel_size=5, stride=1, padding=2)
输入: 1 × 28 ×...
more...