From 66d6525ccfb128470461b0beb494f971ae245949 Mon Sep 17 00:00:00 2001 From: KangBK0120 Date: Fri, 5 Aug 2022 16:13:35 +0900 Subject: [PATCH] documentation: Fix tensorflow and pytorch supported version --- src/sagemaker/huggingface/processing.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/sagemaker/huggingface/processing.py b/src/sagemaker/huggingface/processing.py index 1daeac9c1f..332148891f 100644 --- a/src/sagemaker/huggingface/processing.py +++ b/src/sagemaker/huggingface/processing.py @@ -68,10 +68,10 @@ def __init__( ``image_uri`` is provided. The current supported version is ``4.4.2``. tensorflow_version (str): TensorFlow version you want to use for executing your model training code. Defaults to ``None``. Required unless - ``pytorch_version`` is provided. The current supported version is ``1.6.0``. + ``pytorch_version`` is provided. The current supported version is ``2.4.1``. pytorch_version (str): PyTorch version you want to use for executing your model training code. Defaults to ``None``. Required unless - ``tensorflow_version`` is provided. The current supported version is ``2.4.1``. + ``tensorflow_version`` is provided. The current supported version is ``1.6.0``. py_version (str): Python version you want to use for executing your model training code. Defaults to ``None``. Required unless ``image_uri`` is provided. If using PyTorch, the current supported version is ``py36``. If using TensorFlow,