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
Copy file name to clipboardExpand all lines: doc/using_pytorch.rst
+4-5Lines changed: 4 additions & 5 deletions
Display the source diff
Display the rich diff
Original file line number
Diff line number
Diff line change
@@ -125,8 +125,7 @@ Using third-party libraries
125
125
When running your training script on SageMaker, it will have access to some pre-installed third-party libraries including ``torch``, ``torchvisopm``, and ``numpy``.
126
126
For more information on the runtime environment, including specific package versions, see `SageMaker PyTorch Docker containers <#id4>`__.
127
127
128
-
If there are other packages you want to use with your script, you can include a ``requirements.txt`` file in the same directory as your training script to install other dependencies at runtime.
129
-
A ``requirements.txt`` file is a text file that contains a list of items that are installed by using ``pip install``. You can also specify the version of an item to install.
128
+
If there are other packages you want to use with your script, you can include a ``requirements.txt`` file in the same directory as your training script to install other dependencies at runtime. Both ``requirements.txt`` and your training script should be put in the same folder. You must specify this folder in ``source_dir`` argument when creating PyTorch estimator. A ``requirements.txt`` file is a text file that contains a list of items that are installed by using ``pip install``. You can also specify the version of an item to install.
130
129
For information about the format of a ``requirements.txt`` file, see `Requirements Files <https://pip.pypa.io/en/stable/user_guide/#requirements-files>`__ in the pip documentation.
131
130
132
131
@@ -524,7 +523,7 @@ The PyTorchModel constructor takes the following arguments:
524
523
which should be executed as the entry point to model hosting.
525
524
- ``source_dir:`` Optional. Path (absolute or relative) to a
526
525
directory with any other training source code dependencies including
527
-
tne entry point file. Structure within this directory will be
526
+
the entry point file. Structure within this directory will be
528
527
preserved when training on SageMaker.
529
528
- ``enable_cloudwatch_metrics:`` Optional. If true, training
530
529
and hosting containers will generate Cloudwatch metrics under the
@@ -655,9 +654,9 @@ The following are optional arguments. When you create a ``PyTorch`` object, you
655
654
and where to find the source code to build your custom image.
656
655
657
656
***********************************
658
-
SageMaker PyTorch Docker Containers
657
+
SageMaker PyTorch Docker Containers
659
658
***********************************
660
659
661
660
For information about SageMaker PyTorch containers, see `the SageMaker PyTorch containers repository <https://github.com/aws/sagemaker-pytorch-container>`_.
662
661
663
-
For information about SageMaker PyTorch container dependencies, see `SageMaker PyTorch Containers <https://github.com/aws/sagemaker-python-sdk/tree/master/src/sagemaker/pytorch#sagemaker-pytorch-docker-containers>`_.
662
+
For information about SageMaker PyTorch container dependencies, see `SageMaker PyTorch Containers <https://github.com/aws/sagemaker-python-sdk/tree/master/src/sagemaker/pytorch#sagemaker-pytorch-docker-containers>`_.
0 commit comments