We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent add9443 commit 29baf63Copy full SHA for 29baf63
src/sagemaker/pytorch/estimator.py
@@ -75,7 +75,7 @@ def __init__(
75
framework_version (str): PyTorch version you want to use for
76
executing your model training code. Defaults to ``None``. Required unless
77
``image_uri`` is provided. List of supported versions:
78
- https://github.com/aws/sagemaker-python-sdk#pytorch-sagemaker-estimators.
+ https://github.com/aws/deep-learning-containers/blob/master/available_images.md.
79
py_version (str): Python version you want to use for executing your
80
model training code. One of 'py2' or 'py3'. Defaults to ``None``. Required
81
unless ``image_uri`` is provided.
0 commit comments