Skip to content

Commit 29baf63

Browse files
Documentation: replacing unhelpful link (#2309)
The link that was previously used here does not contain information about supported framework version of PyTorch. Adding a link to the DLC images, which contains all supported PT version. https://github.com/aws/deep-learning-containers/blob/master/available_images.md Co-authored-by: Ahsan Khan <[email protected]>
1 parent add9443 commit 29baf63

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/sagemaker/pytorch/estimator.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -75,7 +75,7 @@ def __init__(
7575
framework_version (str): PyTorch version you want to use for
7676
executing your model training code. Defaults to ``None``. Required unless
7777
``image_uri`` is provided. List of supported versions:
78-
https://github.com/aws/sagemaker-python-sdk#pytorch-sagemaker-estimators.
78+
https://github.com/aws/deep-learning-containers/blob/master/available_images.md.
7979
py_version (str): Python version you want to use for executing your
8080
model training code. One of 'py2' or 'py3'. Defaults to ``None``. Required
8181
unless ``image_uri`` is provided.

0 commit comments

Comments
 (0)