deepspeedai / deepspeedai/DeepSpeedExamples
[ERROR]In Step3，load reward Model failed which trainged with zero-stage 3
Open
@jeffra is already working on this.
Since May 1, 2023.
deespeed chat
new-config
- Dominant language
- Python
- Stars
- 6.8k
- Forks
- 1.1k
- Avg merge
- 2d 16h
- Merged PRs (30d)
- 1
Description
In step 3, I modified the zero-stage of the critic model and reward model from 0 to 3, and the Reward Model trained with zero-stage 3 in step 2 cannot be loaded。But we can load reward model with zero-stage 0 in rw_eval.py
err：size mismatch for rwtranrsformer.layers.0.self_attn.q_proj.weight: copying a param with shape torch.Size([4096, 4096]) from checkpoint, the shape in current model is torch.Size([0]).
modified the zero-stage of the critic model and reward model from 0 to 3：
Contributor guide
No contributing guide indexed for this repository
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Assessment
This issue has not been assessed yet.