Skip to content

Commit 54babf6

Browse files
author
Jonathan Huang
authored
Merge pull request #3123 from tombstone/master
update supported tf versions for object detection on gcloud.
2 parents c0cd713 + 93e964e commit 54babf6

File tree

2 files changed

+9
-1
lines changed

2 files changed

+9
-1
lines changed

research/object_detection/g3doc/running_on_cloud.md

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@ job using GPUs. A sample YAML file is given below:
4242

4343
```
4444
trainingInput:
45-
runtimeVersion: "1.0"
45+
runtimeVersion: "1.2"
4646
scaleTier: CUSTOM
4747
masterType: standard_gpu
4848
workerCount: 9
@@ -71,6 +71,7 @@ following command:
7171
``` bash
7272
# From tensorflow/models/research/
7373
gcloud ml-engine jobs submit training object_detection_`date +%s` \
74+
--runtime-version 1.2 \
7475
--job-dir=gs://${TRAIN_DIR} \
7576
--packages dist/object_detection-0.1.tar.gz,slim/dist/slim-0.1.tar.gz \
7677
--module-name object_detection.train \
@@ -90,6 +91,8 @@ Google Cloud Storage.
9091
Users can monitor the progress of their training job on the [ML Engine
9192
Dashboard](https://console.cloud.google.com/mlengine/jobs).
9293

94+
Note: This sample is supported for use with 1.2 runtime version.
95+
9396
## Running an Evaluation Job on Cloud
9497

9598
Evaluation jobs run on a single machine, so it is not necessary to write a YAML
@@ -98,6 +101,7 @@ job:
98101

99102
``` bash
100103
gcloud ml-engine jobs submit training object_detection_eval_`date +%s` \
104+
--runtime-version 1.2 \
101105
--job-dir=gs://${TRAIN_DIR} \
102106
--packages dist/object_detection-0.1.tar.gz,slim/dist/slim-0.1.tar.gz \
103107
--module-name object_detection.eval \

research/object_detection/g3doc/running_pets.md

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -203,12 +203,15 @@ For running the training Cloud ML job, we'll configure the cluster to use 10
203203
training jobs (1 master + 9 workers) and three parameters servers. The
204204
configuration file can be found at `object_detection/samples/cloud/cloud.yml`.
205205

206+
Note: This sample is supported for use with 1.2 runtime version.
207+
206208
To start training, execute the following command from the
207209
`tensorflow/models/research/` directory:
208210

209211
``` bash
210212
# From tensorflow/models/research/
211213
gcloud ml-engine jobs submit training `whoami`_object_detection_`date +%s` \
214+
--runtime-version 1.2 \
212215
--job-dir=gs://${YOUR_GCS_BUCKET}/train \
213216
--packages dist/object_detection-0.1.tar.gz,slim/dist/slim-0.1.tar.gz \
214217
--module-name object_detection.train \
@@ -224,6 +227,7 @@ Once training has started, we can run an evaluation concurrently:
224227
``` bash
225228
# From tensorflow/models/research/
226229
gcloud ml-engine jobs submit training `whoami`_object_detection_eval_`date +%s` \
230+
--runtime-version 1.2 \
227231
--job-dir=gs://${YOUR_GCS_BUCKET}/train \
228232
--packages dist/object_detection-0.1.tar.gz,slim/dist/slim-0.1.tar.gz \
229233
--module-name object_detection.eval \

0 commit comments

Comments
 (0)