Fix SageMaker local gpu training instructions (#783)

huggingface · Nov 22, 2023 · ccaa63c · ccaa63c
1 parent 6f0fa95
commit ccaa63c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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