Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Fix bug with number of evaluation steps (EleutherAI#384)
we were running way to many evaluation steps if the model is pipe parallel + has g.a.s on because of this line ```python for _ in range(neox_args.gradient_accumulation_steps): ``` - fixing this to 1 if the model is pipe parallel fixes the issue, as .eval_batch() already takes gradient accumulation steps into account.
- Loading branch information