Commit d067ce0a authored by Eli Bixby's avatar Eli Bixby
Browse files

Fix readme + argparse

parent 30fa4ebb
...@@ -73,8 +73,7 @@ $ python cifar10_main.py --data-dir=/prefix/to/downloaded/data/cifar-10-batches- ...@@ -73,8 +73,7 @@ $ python cifar10_main.py --data-dir=/prefix/to/downloaded/data/cifar-10-batches-
--is-cpu-ps=False \ --is-cpu-ps=False \
--force-gpu-compatible=True \ --force-gpu-compatible=True \
--num-gpus=2 \ --num-gpus=2 \
--train-steps=1000
# There are more command line flags to play with; check cifar10_main.py for details. # There are more command line flags to play with; check cifar10_main.py for details.
``` ```
...@@ -99,19 +98,16 @@ Then run the following command from the `tutorials/image` directory of this repo ...@@ -99,19 +98,16 @@ Then run the following command from the `tutorials/image` directory of this repo
``` ```
gcloud ml-engine jobs submit training cifarmultigpu \ gcloud ml-engine jobs submit training cifarmultigpu \
--runtime-version 1.2 \ --runtime-version 1.2 \
--staging-bucket $MY_BUCKET \ --job-dir=$MY_BUCKET/model_dirs/cifarmultigpu \
--config cifar10_estimator/job_config.yaml \ --config cifar10_estimator/cmle_config.yaml \
--package-path cifar10_estimator/ \ --package-path cifar10_estimator/ \
--region us-central1 \
--module-name cifar10_estimator.cifar10_main \ --module-name cifar10_estimator.cifar10_main \
-- \ -- \
--data-dir=$MY_BUCKET/cifar-10-batches-py \ --data-dir=$MY_BUCKET/cifar-10-batches-py \
--job-dir=$MY_BUCKLET/model_dirs/cifarmultigpu \
--is-cpu-ps=True \ --is-cpu-ps=True \
--force-gpu-compatible=True \ --force-gpu-compatible=True \
--num-gpus=4 \ --num-gpus=4 \
--train-steps=1000 \ --train-steps=1000
``` ```
......
...@@ -559,12 +559,6 @@ if __name__ == '__main__': ...@@ -559,12 +559,6 @@ if __name__ == '__main__':
If present when running in a distributed environment will run on sync mode.\ If present when running in a distributed environment will run on sync mode.\
""" """
) )
parser.add_argument(
'--num-workers',
type=int,
default=1,
help='Number of workers.'
)
parser.add_argument( parser.add_argument(
'--num-intra-threads', '--num-intra-threads',
type=int, type=int,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment