@@ -57,25 +57,22 @@ def __init__(
57
57
58
58
Args:
59
59
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``.
63
63
entry_point (str): Path (absolute or relative) to the Python source
64
64
file which should be executed as the entry point to training.
65
65
If ``source_dir`` is specified, then ``entry_point``
66
66
must point to a file located at the root of ``source_dir``.
67
67
transformers_version (str): Transformers version you want to use for
68
68
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``.
71
70
tensorflow_version (str): TensorFlow version you want to use for
72
71
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``.
75
73
pytorch_version (str): PyTorch version you want to use for
76
74
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``.
79
76
source_dir (str): Path (absolute, relative or an S3 URI) to a directory
80
77
with any other training source code dependencies aside from the entry
81
78
point file (default: None). If ``source_dir`` is an S3 URI, it must
0 commit comments