diff --git a/distributed/rpc/batch/parameter_server.py b/distributed/rpc/batch/parameter_server.py index babe667ff6..798f3f0097 100644 --- a/distributed/rpc/batch/parameter_server.py +++ b/distributed/rpc/batch/parameter_server.py @@ -129,7 +129,7 @@ def run(rank, world_size): options=rpc.TensorPipeRpcBackendOptions( num_worker_threads=16, - rpc_timeout=60 # 60 second timeout instead of infinite + rpc_timeout=60 ) if rank != 0: rpc.init_rpc(