m53297601
/
Paddle
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

update worker_num for MPISymetricRoleMaker (#20798)

Browse Source 
test=develop
revert-21172-masked_select_api
Dong Daxiang 5 years ago committed by GitHub
parent 01a9646323
commit ccbdd7aad0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File

3
python/paddle/fluid/incubate/fleet/base/role_maker.py
Unescape View File

@ -252,8 +252,7 @@ class MPISymetricRoleMaker(MPIRoleMaker):
return the current number of worker return the current number of worker
""" """
if self._check_role_generation(): 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 return 0
def _server_num(self): def _server_num(self):

Write Preview
Loading…
Cancel
Save