diff --git a/python/paddle/distributed/fleet/runtime/the_one_ps.py b/python/paddle/distributed/fleet/runtime/the_one_ps.py index 37d79abbab08ea3866141c2636d47dad01ed4830..e0caae9a2fd90d0a861993d6664e4b45f9f137bc 100644 --- a/python/paddle/distributed/fleet/runtime/the_one_ps.py +++ b/python/paddle/distributed/fleet/runtime/the_one_ps.py @@ -165,11 +165,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: