mirror of https://github.com/InternLM/InternLM
remove useless code for no-pp
parent
0fa1083780
commit
54e561665e
|
@ -202,10 +202,8 @@ class NonPipelineScheduler(BaseScheduler):
|
|||
if return_output_label:
|
||||
outputs.append(_output)
|
||||
labels.append(_label)
|
||||
|
||||
if not return_output_label:
|
||||
outputs, labels = None, None
|
||||
|
||||
# Compatible for non-moe
|
||||
if hasattr(gpc.config.model, "num_experts"):
|
||||
return outputs, labels, loss, moe_loss
|
||||
|
|
Loading…
Reference in New Issue