Skip to content

Commit 3b09d94

Browse files
nelyahuloadams
andauthored
fix pipeline eval_batch micro_batches argument for schedule (#6484)
Co-authored-by: Logan Adams <114770087+loadams@users.noreply.github.com>
1 parent 4f80385 commit 3b09d94

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

deepspeed/runtime/pipe/engine.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -482,9 +482,7 @@ def eval_batch(self,
482482
micro_batches = self.micro_batches if num_micro_batches is None else num_micro_batches
483483

484484
# Do the work
485-
sched = schedule.InferenceSchedule(micro_batches=self.micro_batches,
486-
stages=self.num_stages,
487-
stage_id=self.stage_id)
485+
sched = schedule.InferenceSchedule(micro_batches=micro_batches, stages=self.num_stages, stage_id=self.stage_id)
488486

489487
# prevent dead-lock with multiple evals sequence
490488
dist.barrier()

0 commit comments

Comments
 (0)