From aa3840fc3853185b03213fe43459f19d5fb80d53 Mon Sep 17 00:00:00 2001
From: yingtongxiong <974106207@qq.com>
Date: Thu, 26 Oct 2023 20:42:24 +0800
Subject: [PATCH] fix some bugs

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

diff --git a/internlm/train/training_internlm.py b/internlm/train/training_internlm.py
index a4b2e59..2b80692 100644
--- a/internlm/train/training_internlm.py
+++ b/internlm/train/training_internlm.py
@@ -624,4 +624,4 @@ def record_current_batch_training_metrics(
             for tf in tflops_list_2.copy():
                 if abs(tf - avg_tflops_2) > 10:
                     tflops_list_2.remove(tf)
-            print(f"avg_tflops: {sum(tflops_list_2)/len(tflops_list_2)}", flush=True)
+            print(f"avg_tflops_2: {sum(tflops_list_2)/len(tflops_list_2)}", flush=True)