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

fix(no_pp_scheduler): drop out and label if not used

pull/39/head
Sun Peng 2023-07-08 17:08:19 +08:00
parent dfb2751f00
commit 0bfb937710
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
internlm/core/no_pipeline_scheduler.py
View File

@ -269,7 +269,7 @@ class NonPipelineScheduler(BaseScheduler):
if return_loss: if return_loss:
loss += _loss loss += _loss
if return_output_label:
outputs.append(_output) outputs.append(_output)
labels.append(_label) labels.append(_label)