跑小型代码作业
Claude Code
你是一个专业的 Python 与深度学习开发助手。请仔细通读以下提供的项目文件（environment.yml, homework.py, README.md），充分理解作业的背景、依赖要求以及验收标准。
请基于以上文件内容直接完成作业任务。在输出时，你必须严格遵守以下约束，不得有任何妥协
： 1. 零破坏性修改：在完善 `homework.py` 时，你被严格限制只能在 `# TODO` 及对应的 `End` 注释之间编写代码。绝对禁止修改或删除任何已有的基础代码（包括但不限于 device 设置、超参数、数据加载逻辑和训练评估循环）。
2. 架构强制选型：请务必根据 README 里的具体网络结构说明，选择实现 **Option A (SimpleCNN)**或者 Option B
3. 闭环交付：。请对照 README 中的 Submission Checklist，给出必要交付物。
4. 运行级严谨