@@ -57,25 +57,22 @@ def __init__(
5757
5858 Args:
5959 py_version (str): Python version you want to use for executing your model training
60- code. Defaults to ``None``. Required unless ``image_uri`` is provided. List
61- of supported versions:
62- https://github.com/aws/sagemaker-python-sdk#huggingface-sagemaker-estimators
60+ code. Defaults to ``None``. Required unless ``image_uri`` is provided. If
61+ using PyTorch, the current supported version is ``py36``. If using TensorFlow,
62+ the current supported version is ``py37``.
6363 entry_point (str): Path (absolute or relative) to the Python source
6464 file which should be executed as the entry point to training.
6565 If ``source_dir`` is specified, then ``entry_point``
6666 must point to a file located at the root of ``source_dir``.
6767 transformers_version (str): Transformers version you want to use for
6868 executing your model training code. Defaults to ``None``. Required unless
69- ``image_uri`` is provided. List of supported versions:
70- https://github.com/aws/sagemaker-python-sdk#huggingface-sagemaker-estimators.
69+ ``image_uri`` is provided. The current supported version is ``4.4.2``.
7170 tensorflow_version (str): TensorFlow version you want to use for
7271 executing your model training code. Defaults to ``None``. Required unless
73- ``pytorch_version`` is provided. List of supported versions:
74- https://github.com/aws/sagemaker-python-sdk#huggingface-sagemaker-estimators.
72+ ``pytorch_version`` is provided. The current supported version is ``1.6.0``.
7573 pytorch_version (str): PyTorch version you want to use for
7674 executing your model training code. Defaults to ``None``. Required unless
77- ``tensorflow_version`` is provided. List of supported versions:
78- https://github.com/aws/sagemaker-python-sdk#huggingface-sagemaker-estimators.
75+ ``tensorflow_version`` is provided. The current supported version is ``2.4.1``.
7976 source_dir (str): Path (absolute, relative or an S3 URI) to a directory
8077 with any other training source code dependencies aside from the entry
8178 point file (default: None). If ``source_dir`` is an S3 URI, it must
0 commit comments