From 54e561665eb65f0212686051c943f73fd98c716f Mon Sep 17 00:00:00 2001 From: yingtongxiong <974106207@qq.com> Date: Mon, 9 Oct 2023 18:08:15 +0800 Subject: [PATCH] remove useless code for no-pp --- internlm/core/scheduler/no_pipeline_scheduler.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/internlm/core/scheduler/no_pipeline_scheduler.py b/internlm/core/scheduler/no_pipeline_scheduler.py index 9768790..6777acc 100644 --- a/internlm/core/scheduler/no_pipeline_scheduler.py +++ b/internlm/core/scheduler/no_pipeline_scheduler.py @@ -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