github
/
InternLM
mirror of https://github.com/InternLM/InternLM
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

fix(metric): argument missing in getting loss metrics. (#256)

pull/260/head
Ryan (张磊) 2023-08-31 17:44:39 +08:00 committed by GitHub
parent ba4deaeffa
commit c92aa06bd8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
internlm/model/metrics.py
View File

@ -176,7 +176,7 @@ class AccPerplex:
res.update(ds_acc)
res.update(ds_tokens)
loss_res = self.loss_with_type_id.get_metric()
loss_res = self.loss_with_type_id.get_metric(reset)
res.update(loss_res)
return res