diff --git a/intermediate_source/rpc_param_server_tutorial.rst b/intermediate_source/rpc_param_server_tutorial.rst index 5531c51c0f8..324331646c5 100644 --- a/intermediate_source/rpc_param_server_tutorial.rst +++ b/intermediate_source/rpc_param_server_tutorial.rst @@ -310,12 +310,12 @@ We've now completed our trainer and parameter server specific code, and all that help="""Total number of participating processes. Should be the sum of master node and all training nodes.""") parser.add_argument( - "rank", + "--rank", type=int, default=None, help="Global rank of this process. Pass in 0 for master.") parser.add_argument( - "num_gpus", + "--num_gpus", type=int, default=0, help="""Number of GPUs to use for training, Currently supports between 0