diff --git a/docs/sagemaker/source/tutorials/sagemaker-sdk/training-sagemaker-sdk.md b/docs/sagemaker/source/tutorials/sagemaker-sdk/training-sagemaker-sdk.md index 38251be1b..86f48e4c7 100644 --- a/docs/sagemaker/source/tutorials/sagemaker-sdk/training-sagemaker-sdk.md +++ b/docs/sagemaker/source/tutorials/sagemaker-sdk/training-sagemaker-sdk.md @@ -127,7 +127,7 @@ hyperparameters={'epochs': 1, huggingface_estimator = HuggingFace( entry_point='train.py', source_dir='./scripts', - instance_type='ml.p3.2xlarge', + instance_type='ml.g6.12xlarge', instance_count=1, role=role, transformers_version='4.26', @@ -269,7 +269,7 @@ hyperparameters={'epochs': 1, huggingface_estimator = HuggingFace( entry_point='train.py', source_dir='./scripts', - instance_type='ml.p3.2xlarge', + instance_type='ml.g6.12xlarge', instance_count=1, checkpoint_s3_uri=f's3://{sess.default_bucket()}/checkpoints' use_spot_instances=True, @@ -307,7 +307,7 @@ huggingface_estimator = HuggingFace( entry_point='run_glue.py', source_dir='./examples/pytorch/text-classification', git_config=git_config, - instance_type='ml.p3.2xlarge', + instance_type='ml.g6.12xlarge', instance_count=1, role=role, transformers_version='4.26', @@ -333,7 +333,7 @@ metric_definitions = [ huggingface_estimator = HuggingFace( entry_point='train.py', source_dir='./scripts', - instance_type='ml.p3.2xlarge', + instance_type='ml.g6.12xlarge', instance_count=1, role=role, transformers_version='4.26',