diff --git a/python/paddle/fluid/incubate/fleet/base/role_maker.py b/python/paddle/fluid/incubate/fleet/base/role_maker.py index 03657ada029e582a02be713cd1dadee82d0c562e..2471748b4e46ebcc265dd13407effb0d18d4e000 100644 --- a/python/paddle/fluid/incubate/fleet/base/role_maker.py +++ b/python/paddle/fluid/incubate/fleet/base/role_maker.py @@ -252,8 +252,7 @@ class MPISymetricRoleMaker(MPIRoleMaker): return the current number of worker """ if self._check_role_generation(): - if self.is_worker(): - return self._get_size() / self._proc_per_node + return self._get_size() / self._proc_per_node return 0 def _server_num(self):