From c92aa06bd8a35e203cb5ed0db1712d583ad8b38f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ryan=20=28=E5=BC=A0=E7=A3=8A=29?= <MagicDevil.Zhang@qq.com>
Date: Thu, 31 Aug 2023 17:44:39 +0800
Subject: [PATCH] fix(metric): argument missing in getting loss metrics. (#256)

---
 internlm/model/metrics.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/internlm/model/metrics.py b/internlm/model/metrics.py
index 1749aa2..24ce592 100644
--- a/internlm/model/metrics.py
+++ b/internlm/model/metrics.py
@@ -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