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__(
108
108
framework_version (str): Chainer version you want to use for
109
109
executing your model training code. Defaults to ``None``. Required unless
110
110
``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 .
112
112
image_uri (str): If specified, the estimator will use this image
113
113
for training and hosting, instead of selecting the appropriate
114
114
SageMaker official image based on framework_version and
Original file line number Diff line number Diff line change @@ -138,12 +138,10 @@ def __init__(
138
138
unless ``image_uri`` is provided.
139
139
tensorflow_version (str): TensorFlow version you want to use for
140
140
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``.
143
142
pytorch_version (str): PyTorch version you want to use for
144
143
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``.
147
145
py_version (str): Python version you want to use for executing your
148
146
model training code. Defaults to ``None``. Required unless
149
147
``image_uri`` is provided.
Original file line number Diff line number Diff line change @@ -77,7 +77,7 @@ def __init__(
77
77
framework_version (str): MXNet version you want to use for executing
78
78
your model training code. Defaults to `None`. Required unless
79
79
``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/ .
81
81
py_version (str): Python version you want to use for executing your
82
82
model training code. One of 'py2' or 'py3'. Defaults to ``None``. Required
83
83
unless ``image_uri`` is provided.
Original file line number Diff line number Diff line change @@ -75,7 +75,7 @@ def __init__(
75
75
framework_version (str): Scikit-learn version you want to use for
76
76
executing your model training code. Defaults to ``None``. Required
77
77
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
79
79
py_version (str): Python version you want to use for executing your
80
80
model training code (default: 'py3'). Currently, 'py3' is the only
81
81
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__(
59
59
framework_version (str): TensorFlow version you want to use for executing your model
60
60
training code. Defaults to ``None``. Required unless ``image_uri`` is provided.
61
61
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/ .
63
63
model_dir (str or PipelineVariable): S3 location where the checkpoint data and models
64
64
can be exported to during training (default: None). It will be passed in the
65
65
training script as one of the command line arguments. If not specified,
You can’t perform that action at this time.
0 commit comments