Skip to content

Commit 5fae532

Browse files
Add default value for the arguments to simplify the commands in tutorial. (#2136)
* Add default value for arguments. * Add default value for num_minibatches_per_task * Update default value according to the comments.
1 parent fc38176 commit 5fae532

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

elasticdl_client/common/args.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -209,7 +209,7 @@ def add_common_params(parser):
209209
"default to master_resource_request",
210210
)
211211
parser.add_argument(
212-
"--num_workers", type=int, help="Number of workers", default=0
212+
"--num_workers", type=int, help="Number of workers", default=2
213213
)
214214
parser.add_argument(
215215
"--worker_resource_request",
@@ -309,7 +309,7 @@ def add_common_params(parser):
309309
"--num_minibatches_per_task",
310310
type=int,
311311
help="The number of minibatches per task",
312-
required=True,
312+
default=8,
313313
)
314314
parser.add_argument(
315315
"--cluster_spec",
@@ -364,7 +364,7 @@ def add_common_args_between_master_and_worker(parser):
364364
"--minibatch_size",
365365
help="Minibatch size for worker",
366366
type=int,
367-
required=True,
367+
default=64,
368368
)
369369
parser.add_argument(
370370
"--model_zoo",
@@ -455,7 +455,7 @@ def add_common_args_between_master_and_worker(parser):
455455
DistributionStrategy.PARAMETER_SERVER,
456456
DistributionStrategy.ALLREDUCE,
457457
],
458-
default="",
458+
default=DistributionStrategy.PARAMETER_SERVER,
459459
help="Master will use a distribution policy on a list of devices "
460460
"according to the distributed strategy, "
461461
'e.g. "ParameterServerStrategy" or "AllreduceStrategy" or "Local"',

0 commit comments

Comments
 (0)