单通道图片:流程概览
环境与随机性:统一库、字体、随机种子,避免运行差异。
数据准备:transform → datasets.MNIST → DataLoader,按批处理。
模型:展平 28×28 图像,堆叠线性层 + ReLU 得到 logits。
训练 / 测试:独立函数负责日志、评估、绘图。
复用:参数与模块分离,替换组件即可开启新实验

1. 数据准备(MNIST)

2. 定义基础 MLP 模型

3. 测试与可视化工具



4. 训练主循环

5. 执行训练

6. 展平操作与 batch 维度
1. 数据准备(CIFAR-10)

2. 定义彩色版 MLP
