We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 4f80385 commit 3b09d94Copy full SHA for 3b09d94
deepspeed/runtime/pipe/engine.py
@@ -482,9 +482,7 @@ def eval_batch(self,
482
micro_batches = self.micro_batches if num_micro_batches is None else num_micro_batches
483
484
# Do the work
485
- sched = schedule.InferenceSchedule(micro_batches=self.micro_batches,
486
- stages=self.num_stages,
487
- stage_id=self.stage_id)
+ sched = schedule.InferenceSchedule(micro_batches=micro_batches, stages=self.num_stages, stage_id=self.stage_id)
488
489
# prevent dead-lock with multiple evals sequence
490
dist.barrier()
0 commit comments