单通道图片:流程概览
环境与随机性:统一库、字体、随机种子,避免运行差异。
数据准备:transform → datasets.MNIST → DataLoader,按批处理。
模型:展平 28×28 图像,堆叠线性层 + ReLU 得到 logits。
训练 / 测试:独立函数负责日志、评估、绘图。
复用:参数与模块分离,替换组件即可开启新实验
1. 数据准备(MNIST)
2. 定义基础 MLP 模型
3. 测试与可视化工具
4. 训练主循环
5. 执行训练
6. 展平操作与 batch 维度
1. 数据准备(CIFAR-10)
2. 定义彩色版 MLP
@浙大疏锦行