You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Where the s3 url is a path to your training data, within Amazon S3. The constructor keyword arguments define how SageMaker runs your training script and are discussed, in detail, in a later section.
27
+
Where the S3 url is a path to your training data, within Amazon S3. The constructor keyword arguments define how SageMaker runs your training script and are discussed, in detail, in a later section.
28
28
29
29
In the following sections, we'll discuss how to prepare a training script for execution on SageMaker, then how to run that script on SageMaker using an ``MXNet`` Estimator.
30
30
@@ -207,6 +207,16 @@ If you were previously relying on the default save method, you can now import on
207
207
208
208
save(args.model_dir, model)
209
209
210
+
Lastly, if you were relying on the container launching a parameter server for use with distributed training, you must now set ``launch_parameter_server`` to ``True`` when creating an MXNet estimator:
@@ -218,7 +228,9 @@ If there are other packages you want to use with your script, you can include a
218
228
Running an MXNet training script in SageMaker
219
229
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
220
230
221
-
You run MXNet training scripts on SageMaker by creating ``MXNet`` Estimators. SageMaker training of your script is invoked when you call ``fit`` on an ``MXNet`` Estimator. The following code sample shows how you train a custom MXNet script "train.py".
231
+
You run MXNet training scripts on SageMaker by creating an ``MXNet`` estimators.
232
+
When you call ``fit`` on an ``MXNet`` estimator, a SageMaker training job with your script is started.
233
+
The following code sample shows how you train a custom MXNet script "train.py".
222
234
223
235
.. code:: python
224
236
@@ -292,6 +304,8 @@ The following are optional arguments. When you create an ``MXNet`` object, you c
292
304
framework_version and py_version. Refer to: `SageMaker MXNet Docker Containers
293
305
<#sagemaker-mxnet-docker-containers>`_ for details on what the Official images support
294
306
and where to find the source code to build your custom image.
307
+
- ``launch_parameter_server`` For versions 1.3 and above only.
308
+
This determines whether or not a parameter server is launched for use with distributed training.
0 commit comments