Skip to content

Commit d708751

Browse files
authored
Merge branch 'master' into local_mode_timeout
2 parents d314d7a + cecea12 commit d708751

File tree

4 files changed

+7
-6
lines changed

4 files changed

+7
-6
lines changed

CHANGELOG.rst

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,9 +2,10 @@
22
CHANGELOG
33
=========
44

5-
1.16.2dev
5+
1.16.2
66
======
77

8+
* doc-fix: Change ``distribution`` to ``distributions``
89
* bug-fix: Increase docker-compose http timeout and health check timeout to 120.
910

1011
1.16.1.post1

src/sagemaker/mxnet/README.rst

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -209,15 +209,15 @@ If you were previously relying on the default save method, you can now import on
209209
210210
save(args.model_dir, model)
211211
212-
Lastly, if you were relying on the container launching a parameter server for use with distributed training, you must now set ``distribution`` to the following dictionary when creating an MXNet estimator:
212+
Lastly, if you were relying on the container launching a parameter server for use with distributed training, you must now set ``distributions`` to the following dictionary when creating an MXNet estimator:
213213

214214
.. code:: python
215215
216216
from sagemaker.mxnet import MXNet
217217
218218
estimator = MXNet('path-to-distributed-training-script.py',
219219
...,
220-
distribution={'parameter_server': {'enabled': True}})
220+
distributions={'parameter_server': {'enabled': True}})
221221
222222
223223
Using third-party libraries
@@ -323,7 +323,7 @@ The following are optional arguments. When you create an ``MXNet`` object, you c
323323
framework_version and py_version. Refer to: `SageMaker MXNet Docker Containers
324324
<#sagemaker-mxnet-docker-containers>`_ for details on what the Official images support
325325
and where to find the source code to build your custom image.
326-
- ``distribution`` For versions 1.3 and above only.
326+
- ``distributions`` For versions 1.3 and above only.
327327
Specifies information for how to run distributed training.
328328
To launch a parameter server during training, set this argument to:
329329

src/sagemaker/mxnet/estimator.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -67,7 +67,7 @@ def __init__(self, entry_point, source_dir=None, hyperparameters=None, py_versio
6767
Examples:
6868
123.dkr.ecr.us-west-2.amazonaws.com/my-custom-image:1.0
6969
custom-image:latest.
70-
distribution (dict): A dictionary with information on how to run distributed training
70+
distributions (dict): A dictionary with information on how to run distributed training
7171
(default: None).
7272
**kwargs: Additional kwargs passed to the :class:`~sagemaker.estimator.Framework` constructor.
7373
"""

src/sagemaker/tensorflow/estimator.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -199,7 +199,7 @@ def __init__(self, training_steps=None, evaluation_steps=None, checkpoint_path=N
199199
custom-image:latest.
200200
script_mode (bool): If set to True will the estimator will use the Script Mode containers (default: False).
201201
This will be ignored if py_version is set to 'py3'.
202-
distribution (dict): A dictionary with information on how to run distributed training
202+
distributions (dict): A dictionary with information on how to run distributed training
203203
(default: None). Currently we only support distributed training with parameter servers. To enable it
204204
use the following setup:
205205
{

0 commit comments

Comments
 (0)