Unverified Commit 54babf62 authored by Jonathan Huang's avatar Jonathan Huang Committed by GitHub
Browse files

Merge pull request #3123 from tombstone/master

update supported tf versions for object detection on gcloud.
parents c0cd713f 93e964eb
...@@ -42,7 +42,7 @@ job using GPUs. A sample YAML file is given below: ...@@ -42,7 +42,7 @@ job using GPUs. A sample YAML file is given below:
``` ```
trainingInput: trainingInput:
runtimeVersion: "1.0" runtimeVersion: "1.2"
scaleTier: CUSTOM scaleTier: CUSTOM
masterType: standard_gpu masterType: standard_gpu
workerCount: 9 workerCount: 9
...@@ -71,6 +71,7 @@ following command: ...@@ -71,6 +71,7 @@ following command:
``` bash ``` bash
# From tensorflow/models/research/ # From tensorflow/models/research/
gcloud ml-engine jobs submit training object_detection_`date +%s` \ gcloud ml-engine jobs submit training object_detection_`date +%s` \
--runtime-version 1.2 \
--job-dir=gs://${TRAIN_DIR} \ --job-dir=gs://${TRAIN_DIR} \
--packages dist/object_detection-0.1.tar.gz,slim/dist/slim-0.1.tar.gz \ --packages dist/object_detection-0.1.tar.gz,slim/dist/slim-0.1.tar.gz \
--module-name object_detection.train \ --module-name object_detection.train \
...@@ -90,6 +91,8 @@ Google Cloud Storage. ...@@ -90,6 +91,8 @@ Google Cloud Storage.
Users can monitor the progress of their training job on the [ML Engine Users can monitor the progress of their training job on the [ML Engine
Dashboard](https://console.cloud.google.com/mlengine/jobs). Dashboard](https://console.cloud.google.com/mlengine/jobs).
Note: This sample is supported for use with 1.2 runtime version.
## Running an Evaluation Job on Cloud ## Running an Evaluation Job on Cloud
Evaluation jobs run on a single machine, so it is not necessary to write a YAML Evaluation jobs run on a single machine, so it is not necessary to write a YAML
...@@ -98,6 +101,7 @@ job: ...@@ -98,6 +101,7 @@ job:
``` bash ``` bash
gcloud ml-engine jobs submit training object_detection_eval_`date +%s` \ gcloud ml-engine jobs submit training object_detection_eval_`date +%s` \
--runtime-version 1.2 \
--job-dir=gs://${TRAIN_DIR} \ --job-dir=gs://${TRAIN_DIR} \
--packages dist/object_detection-0.1.tar.gz,slim/dist/slim-0.1.tar.gz \ --packages dist/object_detection-0.1.tar.gz,slim/dist/slim-0.1.tar.gz \
--module-name object_detection.eval \ --module-name object_detection.eval \
......
...@@ -203,12 +203,15 @@ For running the training Cloud ML job, we'll configure the cluster to use 10 ...@@ -203,12 +203,15 @@ For running the training Cloud ML job, we'll configure the cluster to use 10
training jobs (1 master + 9 workers) and three parameters servers. The training jobs (1 master + 9 workers) and three parameters servers. The
configuration file can be found at `object_detection/samples/cloud/cloud.yml`. configuration file can be found at `object_detection/samples/cloud/cloud.yml`.
Note: This sample is supported for use with 1.2 runtime version.
To start training, execute the following command from the To start training, execute the following command from the
`tensorflow/models/research/` directory: `tensorflow/models/research/` directory:
``` bash ``` bash
# From tensorflow/models/research/ # From tensorflow/models/research/
gcloud ml-engine jobs submit training `whoami`_object_detection_`date +%s` \ gcloud ml-engine jobs submit training `whoami`_object_detection_`date +%s` \
--runtime-version 1.2 \
--job-dir=gs://${YOUR_GCS_BUCKET}/train \ --job-dir=gs://${YOUR_GCS_BUCKET}/train \
--packages dist/object_detection-0.1.tar.gz,slim/dist/slim-0.1.tar.gz \ --packages dist/object_detection-0.1.tar.gz,slim/dist/slim-0.1.tar.gz \
--module-name object_detection.train \ --module-name object_detection.train \
...@@ -224,6 +227,7 @@ Once training has started, we can run an evaluation concurrently: ...@@ -224,6 +227,7 @@ Once training has started, we can run an evaluation concurrently:
``` bash ``` bash
# From tensorflow/models/research/ # From tensorflow/models/research/
gcloud ml-engine jobs submit training `whoami`_object_detection_eval_`date +%s` \ gcloud ml-engine jobs submit training `whoami`_object_detection_eval_`date +%s` \
--runtime-version 1.2 \
--job-dir=gs://${YOUR_GCS_BUCKET}/train \ --job-dir=gs://${YOUR_GCS_BUCKET}/train \
--packages dist/object_detection-0.1.tar.gz,slim/dist/slim-0.1.tar.gz \ --packages dist/object_detection-0.1.tar.gz,slim/dist/slim-0.1.tar.gz \
--module-name object_detection.eval \ --module-name object_detection.eval \
......
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