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

2023-07-08 17:08:19 +08:00 · 2023-07-08 17:08:19 +08:00 · 0bfb937710
parent dfb2751f00
commit 0bfb937710
1 changed files with 3 additions and 3 deletions
--- a/internlm/core/no_pipeline_scheduler.py
+++ b/internlm/core/no_pipeline_scheduler.py
@ -269,7 +269,7 @@ class NonPipelineScheduler(BaseScheduler):

            if return_loss:
                loss += _loss
-
+            if return_output_label:
                outputs.append(_output)
                labels.append(_label)