未验证 提交 859431aa 编写于 作者: C Chengmo 提交者: GitHub

fix ps init(#30397)

Co-authored-by: NseiriosPlus <tangwei12@baidu.com>
上级 2a98e932
...@@ -168,11 +168,7 @@ class CommonAccessor: ...@@ -168,11 +168,7 @@ class CommonAccessor:
shape = self.get_shard(total_dims, pserver_num, pserver_id) shape = self.get_shard(total_dims, pserver_num, pserver_id)
dims.append(shape) dims.append(shape)
if formal_name == "Param": initializer = self.get_initializer_attr(param.name, startup_program)
initializer = "uniform_random&0&-1.0&1.0"
else:
initializer = self.get_initializer_attr(param.name,
startup_program)
initializers.append(initializer) initializers.append(initializer)
for (attr_varname, type_) in attr_varnames: for (attr_varname, type_) in attr_varnames:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册