fix ps init(#30397)

Co-authored-by: seiriosPlus <tangwei12@baidu.com>
4 years ago · 859431aadb
parent 2a98e9323a
commit 859431aadb
1 changed files with 1 additions and 5 deletions
--- 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: