@@ -209,7 +209,7 @@ def add_common_params(parser):
209
209
"default to master_resource_request" ,
210
210
)
211
211
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
213
213
)
214
214
parser .add_argument (
215
215
"--worker_resource_request" ,
@@ -309,7 +309,7 @@ def add_common_params(parser):
309
309
"--num_minibatches_per_task" ,
310
310
type = int ,
311
311
help = "The number of minibatches per task" ,
312
- required = True ,
312
+ default = 8 ,
313
313
)
314
314
parser .add_argument (
315
315
"--cluster_spec" ,
@@ -364,7 +364,7 @@ def add_common_args_between_master_and_worker(parser):
364
364
"--minibatch_size" ,
365
365
help = "Minibatch size for worker" ,
366
366
type = int ,
367
- required = True ,
367
+ default = 64 ,
368
368
)
369
369
parser .add_argument (
370
370
"--model_zoo" ,
@@ -455,7 +455,7 @@ def add_common_args_between_master_and_worker(parser):
455
455
DistributionStrategy .PARAMETER_SERVER ,
456
456
DistributionStrategy .ALLREDUCE ,
457
457
],
458
- default = "" ,
458
+ default = DistributionStrategy . PARAMETER_SERVER ,
459
459
help = "Master will use a distribution policy on a list of devices "
460
460
"according to the distributed strategy, "
461
461
'e.g. "ParameterServerStrategy" or "AllreduceStrategy" or "Local"' ,
0 commit comments