ModelCenter icon indicating copy to clipboard operation
ModelCenter copied to clipboard

模型加载问题

Open ftgreat opened this issue 1 year ago • 0 comments

https://github.com/OpenBMB/ModelCenter/blob/main/examples/cpm2/pretrain_cpm2.py#L24

请问这里模型初始化是不是每卡都会执行? 如果模型很大,可能内存OOM。谢谢您的解答。

ftgreat avatar Jul 13 '23 05:07 ftgreat