模型加载问题
Open
Nobody has claimed this yet.
- Dominant language
- Python
- Stars
- 270
- Forks
- 32
- PR merge metrics
- No merged PRs in 30d
Description
https://github.com/OpenBMB/ModelCenter/blob/main/examples/cpm2/pretrain_cpm2.py#L24
请问这里模型初始化是不是每卡都会执行？
如果模型很大，可能内存OOM。谢谢您的解答。
Contributor guide
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.
Research direction
Read examples/cpm2/pretrain_cpm2.py at line 24 and trace the model initialization path through the distributed startup flow. Determine whether initialization occurs independently on each device and document the observed memory behavior; the issue does not define a specific code change or completion test.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python
- Domain
- distributed-systems, machine-learning
- Issue type
- Bug
- Difficulty
- 5/5
- Estimated time
- Over a week
- Activity status
- Stale
- Clarity
- Needs clarification
- Newbie friendliness
- 15/100