Fix SageMaker local gpu training instructions (#783)

sasan-j · web-flow · commit ccaa63cfca80 · 2023-11-22T22:16:50.000+01:00
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
 

Original file line number	Diff line number	Diff line change
`@@ -137,7 +137,7 @@ huggingface_estimator = HuggingFace(`
`137`	`137`	`)`
`138`	`138`	```
`139`	`139`
`140`		-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.
	`140`	+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.
`141`	`141`
`142`	`142`	`## Execute training`
`143`	`143`