Skip to content

Commit fe829d8

Browse files
authored
Update using_tf.rst
Added review changes
1 parent 31c3ce9 commit fe829d8

File tree

1 file changed

+6
-5
lines changed

1 file changed

+6
-5
lines changed

doc/frameworks/tensorflow/using_tf.rst

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -200,22 +200,23 @@ The constructor keyword arguments define how SageMaker runs your training script
200200
Specify a Docker image using an Estimator
201201
-----------------------------------------
202202

203-
You can also specify a Docker image when creating an Estimator by specifying the ECR URI for the Python and framework version directly. For a full list of available container URIs, see `Available Deep Learning Containers Images <https://github.com/aws/deep-learning-containers/blob/master/available_images.md>`__ For more information on using Docker containers, see `Use Your Own Algorithms or Models with Amazon SageMaker <https://docs.aws.amazon.com/sagemaker/latest/dg/your-algorithms.html>`__.
203+
There are use cases, such as extending an existing pre-built Amazon SageMaker images, that require specifing a Docker image when creating an Estimator by directly specifying the ECR URI instead of the Python and framework version. For a full list of available container URIs, see `Available Deep Learning Containers Images <https://github.com/aws/deep-learning-containers/blob/master/available_images.md>`__ For more information on using Docker containers, see `Use Your Own Algorithms or Models with Amazon SageMaker <https://docs.aws.amazon.com/sagemaker/latest/dg/your-algorithms.html>`__.
204204

205-
When specifying the image, you must use the ``image_name=''`` arg to replace both of the following args:
205+
When specifying the image, you must use the ``image_name=''`` arg to replace the following arg:
206206

207207
- ``py_version=''``
208-
- ``framework_version=''``
209208

210-
The following example uses the ``image_name='763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-training:1.15.2-cpu-py37-ubuntu18.04'`` arg to specify the container image, Python version, and framework version:
209+
You should still specify the ``framework_version=''`` arg because the SageMaker Python SDK accomodates for differences in the images based on the version.
210+
211+
The following example uses the ``image_name=''`` arg to specify the container image, Python version, and framework version.
211212

212213
.. code:: python
213214
214215
tf_estimator = TensorFlow(entry_point='tf-train.py',
215216
role='SageMakerRole',
216217
train_instance_count=1,
217218
train_instance_type='ml.p2.xlarge',
218-
image_name='763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-training:1.15.2-cpu-py37-ubuntu18.04',
219+
image_name='763104351884.dkr.ecr.<region>.amazonaws.com/<framework>-<job type>:<framework version>-<cpu/gpu>-<python version>-ubuntu18.04',
219220
script_mode=True)
220221
221222
For more information about the sagemaker.tensorflow.TensorFlow estimator, see `SageMaker TensorFlow Classes`_.

0 commit comments

Comments
 (0)