Skip to content

Commit f4ebb4d

Browse files
oabuhamdansvekars
andauthored
Rank and Num GPUs flags are missing the double dash. Adding those (#2841)
Co-authored-by: Svetlana Karslioglu <[email protected]>
1 parent f748810 commit f4ebb4d

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

intermediate_source/rpc_param_server_tutorial.rst

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -310,12 +310,12 @@ We've now completed our trainer and parameter server specific code, and all that
310310
help="""Total number of participating processes. Should be the sum of
311311
master node and all training nodes.""")
312312
parser.add_argument(
313-
"rank",
313+
"--rank",
314314
type=int,
315315
default=None,
316316
help="Global rank of this process. Pass in 0 for master.")
317317
parser.add_argument(
318-
"num_gpus",
318+
"--num_gpus",
319319
type=int,
320320
default=0,
321321
help="""Number of GPUs to use for training, Currently supports between 0

0 commit comments

Comments
 (0)