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