From 6144ce5ebf47babd513d0e3a38c32e1037735421 Mon Sep 17 00:00:00 2001 From: Sasan Jafarnejad Date: Tue, 16 May 2023 18:45:41 +0200 Subject: [PATCH] Fix SageMaker local gpu training instructions --- 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 690bbe896..c69b4c4ef 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