5 条评论 (5 条评论)0 个反应 (0 个反应)0 位负责人 (0 位负责人)Python9,429 个派生 (9,429 个派生)batch import
good first issue
仓库指标
- 星标
- 21,634 个星标 (21,634 个星标)
- PR 合并指标
- PR 指标待抓取 (PR 指标待抓取)
贡献者指南
- 研究方向
- 该问题指向 word language model/main.py 的第 171 行(https://github.com/pytorch/examples/blob/7d0d413425e2ee64fcd0e0de1b11c5cca1f79f4d/word language model/main.py#L171)。当前代码在每个 epoch 结束时保存模型,但问题指出它保存的是当前模型,而非最佳模型。修复方法是添加逻辑来跟踪最佳验证损失,并仅在达到新的最佳值时保存模型。确保使用 torch.save 正确保存模型状态。更改应局限于该行附近的训练循环。没有关联的 PR 或近期评论表明有正在进行的工作。
- 技术栈
- pythonpytorch
- 领域
- machine learningai
- 议题类型
- 缺陷
- 前置要求
- GitPythonPyTorch