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

fix some bugs

pull/456/head
yingtongxiong 2023-10-26 20:42:24 +08:00
parent 8aefb74e02
commit aa3840fc38
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/train/training_internlm.py
View File

@ -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)