已开启
[fully_shard] 延迟初始化场景下,fully_shard内部报错AllGather不支持metaTensor,且有reset_parameter中不应该setattr parameter到对应module中 #67
MengXY107创建于  3月31日
MengXY107
MengXY107成员
3月31日 创建

该问题是怎么引起的?

在调试Meta初始化的场景下,发现预取配置下,fully_shard内部报错AllGather不支持metaTensor,根因在于root module 要将所有的HSDPState进行lazy_init,
此外,HSDPParam.reset_parameters方法中,当输入是普通Tensor时的处理分支,不应该进行settattr parameter 到对应module中,否则会导致优化器引用的参数和sharded_param不是同一个对象,引发严重的精度问题。

重现步骤

报错信息

likedislike
MengXY107MengXY107成员
3月31日 将 Meng107 设为负责人