File tree Expand file tree Collapse file tree 5 files changed +6
-8
lines changed Expand file tree Collapse file tree 5 files changed +6
-8
lines changed Original file line number Diff line number Diff line change @@ -108,7 +108,7 @@ def __init__(
108108 framework_version (str): Chainer version you want to use for
109109 executing your model training code. Defaults to ``None``. Required unless
110110 ``image_uri`` is provided. List of supported versions:
111- https://github.com/aws/sagemaker-python-sdk# chainer-sagemaker-estimators .
111+ https://sagemaker.readthedocs.io/en/stable/frameworks/chainer/using_chainer.html#using- chainer-with-the- sagemaker-python-sdk .
112112 image_uri (str): If specified, the estimator will use this image
113113 for training and hosting, instead of selecting the appropriate
114114 SageMaker official image based on framework_version and
Original file line number Diff line number Diff line change @@ -138,12 +138,10 @@ def __init__(
138138 unless ``image_uri`` is provided.
139139 tensorflow_version (str): TensorFlow version you want to use for
140140 executing your inference code. Defaults to ``None``. Required unless
141- ``pytorch_version`` is provided. List of supported versions:
142- https://github.com/aws/sagemaker-python-sdk#huggingface-sagemaker-estimators.
141+ ``pytorch_version`` is provided. The current supported version is ``2.4.1``.
143142 pytorch_version (str): PyTorch version you want to use for
144143 executing your inference code. Defaults to ``None``. Required unless
145- ``tensorflow_version`` is provided. List of supported versions:
146- https://github.com/aws/sagemaker-python-sdk#huggingface-sagemaker-estimators.
144+ ``tensorflow_version`` is provided. The current supported versions are ``1.7.1`` and ``1.6.0``.
147145 py_version (str): Python version you want to use for executing your
148146 model training code. Defaults to ``None``. Required unless
149147 ``image_uri`` is provided.
Original file line number Diff line number Diff line change @@ -77,7 +77,7 @@ def __init__(
7777 framework_version (str): MXNet version you want to use for executing
7878 your model training code. Defaults to `None`. Required unless
7979 ``image_uri`` is provided. List of supported versions.
80- https://github. com/aws/sagemaker-python-sdk#mxnet-sagemaker-estimators .
80+ https://aws.amazon. com/releasenotes/available-deep-learning-containers-images/ .
8181 py_version (str): Python version you want to use for executing your
8282 model training code. One of 'py2' or 'py3'. Defaults to ``None``. Required
8383 unless ``image_uri`` is provided.
Original file line number Diff line number Diff line change @@ -75,7 +75,7 @@ def __init__(
7575 framework_version (str): Scikit-learn version you want to use for
7676 executing your model training code. Defaults to ``None``. Required
7777 unless ``image_uri`` is provided. List of supported versions:
78- https://github. com/aws/ sagemaker-python-sdk# sklearn-sagemaker-estimators
78+ https://docs.aws.amazon. com/sagemaker/latest/dg/ sklearn.html
7979 py_version (str): Python version you want to use for executing your
8080 model training code (default: 'py3'). Currently, 'py3' is the only
8181 supported version. If ``None`` is passed in, ``image_uri`` must be
Original file line number Diff line number Diff line change @@ -59,7 +59,7 @@ def __init__(
5959 framework_version (str): TensorFlow version you want to use for executing your model
6060 training code. Defaults to ``None``. Required unless ``image_uri`` is provided.
6161 List of supported versions:
62- https://github. com/aws/sagemaker-python-sdk#tensorflow-sagemaker-estimators .
62+ https://aws.amazon. com/releasenotes/available-deep-learning-containers-images/ .
6363 model_dir (str or PipelineVariable): S3 location where the checkpoint data and models
6464 can be exported to during training (default: None). It will be passed in the
6565 training script as one of the command line arguments. If not specified,
You can’t perform that action at this time.
0 commit comments