Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix NVFLARE demo (#8340)

Browse Source
This commit is contained in:
Rong Ou 2022-10-13 21:18:34 -07:00 committed by GitHub
parent 748d516c50
commit b3208aac4e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
demo/nvflare/custom/trainer.py
View File

@ -53,6 +53,7 @@ class XGBoostTrainer(Executor):
client_name = fl_ctx.get_prop(FLContextKey.CLIENT_NAME) client_name = fl_ctx.get_prop(FLContextKey.CLIENT_NAME)
rank = int(client_name.split('-')[1]) - 1 rank = int(client_name.split('-')[1]) - 1
communicator_env = { communicator_env = {
'xgboost_communicator': 'federated',
'federated_server_address': self._server_address, 'federated_server_address': self._server_address,
'federated_world_size': self._world_size, 'federated_world_size': self._world_size,
'federated_rank': rank, 'federated_rank': rank,