pytorch/examples

Query on loss calculation in word language model

オープン

#1,082 opened on 2022/10/14

 (0 件のコメント) (0 件のリアクション) (0 人の担当者)Python (9,429 件のフォーク)batch import
help wanted

Repository metrics

Stars
 (21,634 個のスター)
PR merge metrics
 (PR metrics pending)

説明

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?

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