pytorch/examples

Query on loss calculation in word language model

Open

#1,082 opened on 2022年10月14日

GitHub で見る
 (0 comments) (0 reactions) (0 assignees)Python (21,634 stars) (9,429 forks)batch import
help wanted

説明

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?

コントリビューターガイド