bigscience-workshop / bigscience-workshop/Megatron-DeepSpeed
Make sure deepspeed powered models are equivalent with their non deepspeed version
- Dominant language
- Python
- Stars
- 1.4k
- Forks
- 226
- PR merge metrics
- No merged PRs in 30d
Description
@DanielHesslow has opened a PR https://github.com/bigscience-workshop/Megatron-DeepSpeed/pull/212. This allows us to evaluate Megatron-Deepspeed models using the EAI harness directly in this repo, without needing to convert models into HF format.
The current issue, is that we train models using deepspeed (*ModelPipe) but the evaluation script loads a model without deepspeed (*Model). This creates issues where we might have discrepancies between those two. Ex: https://github.com/bigscience-workshop/Megatron-DeepSpeed/pull/222
We need a test making sure that the output of both models are equal given an arbitrary configurations (regardless if we merge #212 or not)
cc @SaulLu @DanielHesslow @TevenLeScao
Contributor guide
No contributing guide indexed for this repository
Assessment
This issue has not been assessed yet.