From ccaa63cfca806769f04c9eb351902601cf130505 Mon Sep 17 00:00:00 2001 From: Sasan Jafarnejad Date: Wed, 22 Nov 2023 22:16:50 +0100 Subject: [PATCH] Fix SageMaker local gpu training instructions (#783) --- docs/sagemaker/train.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/sagemaker/train.md b/docs/sagemaker/train.md index cc977e0c3..4fc33466a 100644 --- a/docs/sagemaker/train.md +++ b/docs/sagemaker/train.md @@ -137,7 +137,7 @@ huggingface_estimator = HuggingFace( ) ``` -If you are running a `TrainingJob` locally, define `instance_type='local'` or `instance_type='local-gpu'` for GPU usage. Note that this will not work with SageMaker Studio. +If you are running a `TrainingJob` locally, define `instance_type='local'` or `instance_type='local_gpu'` for GPU usage. Note that this will not work with SageMaker Studio. ## Execute training