fix ps init(#30397)

Co-authored-by: N seiriosPlus <tangwei12@baidu.com>

fix ps init(#30397)
Co-authored-by: N seiriosPlus <tangwei12@baidu.com>
859431aa · Chengmo · GitHub · 2a98e932 · 859431aa
隐藏空白更改
内联并排

Showing with 1 addition and 5 deletion

python/paddle/distributed/fleet/runtime/the_one_ps.py python/paddle/distributed/fleet/runtime/the_one_ps.py +1 -5

未找到文件。
--- a/python/paddle/distributed/fleet/runtime/the_one_ps.py
+++ b/python/paddle/distributed/fleet/runtime/the_one_ps.py
@@ -168,11 +168,7 @@ class CommonAccessor:
                    shape = self.get_shard(total_dims, pserver_num, pserver_id)
            dims.append(shape)

-            if formal_name == "Param":
-                initializer = "uniform_random&0&-1.0&1.0"
-            else:
-                initializer = self.get_initializer_attr(param.name,
-                                                        startup_program)
+            initializer = self.get_initializer_attr(param.name, startup_program)
            initializers.append(initializer)

        for (attr_varname, type_) in attr_varnames: