From bf7bbb947dc864c8b305ccb77a437b940f43094e Mon Sep 17 00:00:00 2001 From: JMGaljaard Date: Mon, 26 Sep 2022 10:53:07 +0200 Subject: [PATCH] Add hotfix for fed learning --- fltk/launch.py | 1 + 1 file changed, 1 insertion(+) diff --git a/fltk/launch.py b/fltk/launch.py index cf374761..dfe030af 100644 --- a/fltk/launch.py +++ b/fltk/launch.py @@ -271,6 +271,7 @@ def launch_remote(arg_path: Path, conf_path: Path, rank: Rank, nic: Optional[NIC msg = f'Starting with host={host} and port={os.environ["MASTER_PORT"]} and interface={nic}' logging.log(logging.INFO, msg) + # fixme: Move to async implementation to prevent CPP memory leak options = rpc.TensorPipeRpcBackendOptions( num_worker_threads=16 if rank == 0 else 4, rpc_timeout=0, # infinite timeout