diff --git a/python/paddle/distributed/fleet/runtime/the_one_ps.py b/python/paddle/distributed/fleet/runtime/the_one_ps.py index d9e4c8cb8decd544fd0e1cfc2bb64b8b109b0846..20bf443689ef06d37287cbcb27392e9c0e137040 100644 --- 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: