pytorch/examples

Query on loss calculation in word language model

Open

#1,082 建立於 2022年10月14日

在 GitHub 查看
 (0 留言) (0 反應) (0 負責人)Python (9,429 fork)batch import
help wanted

倉庫指標

Star
 (21,634 star)
PR 合併指標
 (30 天內沒有已合併 PR)

描述

In the main.py of word language model, I find that in the evaluate function the total_loss is getting multiplied by length of data https://github.com/pytorch/examples/blob/ca1bd9167f7216e087532160fc5b98643d53f87e/word_language_model/main.py#L163

However in the train function, total_loss is not getting multiplied by length of data https://github.com/pytorch/examples/blob/ca1bd9167f7216e087532160fc5b98643d53f87e/word_language_model/main.py#L194

Is this proper?

貢獻者指南

研究方向
檢查 word language model/main.py 第 163 行和第 194 行的 evaluate 和 train 函數。在 evaluate 中，total loss 乘以了 len(data)，但在 train 中沒有。確定這是為了在資料集上求平均而有意為之，還是一個不一致之處。如果是 bug，提出修復方案；否則，加入程式碼註解解釋差異。
技術棧
python
領域
machine learning
議題類型
錯誤
難度面向新貢獻者的預計實作難度，1 表示很小改動，5 表示專家級工作。
2
預計時間有經驗貢獻者完成調查、實作、測試並準備 pull request 的粗略時間範圍。
1 小時以內
活動狀態議題目前的可參與程度：新鮮、活躍、陳舊、阻塞或等待維護者輸入。
新近可參與
清晰度議題是否清楚說明預期改動、驗收標準和下一步。
清晰
前置要求
PyTorch
新手友善度1-100 的估計分數，表示該議題對首次貢獻者的友善程度。
75