update worker_num for MPISymetricRoleMaker (#20798)

test=develop
5 years ago · ccbdd7aad0
parent 01a9646323
commit ccbdd7aad0
1 changed files with 1 additions and 2 deletions
--- 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):