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

remove useless code for no-pp

pull/407/head
yingtongxiong 2023-10-09 18:08:15 +08:00
parent 0fa1083780
commit 54e561665e
1 changed files with 0 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
internlm/core/scheduler/no_pipeline_scheduler.py
View File

@ -202,10 +202,8 @@ class NonPipelineScheduler(BaseScheduler):
if return_output_label: if return_output_label:
outputs.append(_output) outputs.append(_output)
labels.append(_label) labels.append(_label)
if not return_output_label: if not return_output_label:
outputs, labels = None, None outputs, labels = None, None
# Compatible for non-moe # Compatible for non-moe
if hasattr(gpc.config.model, "num_experts"): if hasattr(gpc.config.model, "num_experts"):
return outputs, labels, loss, moe_loss return outputs, labels, loss, moe_loss