Skip to content

Commit 4323440

Browse files
authored
instance change (#1975)
1 parent 5bb267a commit 4323440

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

docs/sagemaker/source/tutorials/sagemaker-sdk/training-sagemaker-sdk.md

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -127,7 +127,7 @@ hyperparameters={'epochs': 1,
127127
huggingface_estimator = HuggingFace(
128128
entry_point='train.py',
129129
source_dir='./scripts',
130-
instance_type='ml.p3.2xlarge',
130+
instance_type='ml.g6.12xlarge',
131131
instance_count=1,
132132
role=role,
133133
transformers_version='4.26',
@@ -269,7 +269,7 @@ hyperparameters={'epochs': 1,
269269
huggingface_estimator = HuggingFace(
270270
entry_point='train.py',
271271
source_dir='./scripts',
272-
instance_type='ml.p3.2xlarge',
272+
instance_type='ml.g6.12xlarge',
273273
instance_count=1,
274274
checkpoint_s3_uri=f's3://{sess.default_bucket()}/checkpoints'
275275
use_spot_instances=True,
@@ -307,7 +307,7 @@ huggingface_estimator = HuggingFace(
307307
entry_point='run_glue.py',
308308
source_dir='./examples/pytorch/text-classification',
309309
git_config=git_config,
310-
instance_type='ml.p3.2xlarge',
310+
instance_type='ml.g6.12xlarge',
311311
instance_count=1,
312312
role=role,
313313
transformers_version='4.26',
@@ -333,7 +333,7 @@ metric_definitions = [
333333
huggingface_estimator = HuggingFace(
334334
entry_point='train.py',
335335
source_dir='./scripts',
336-
instance_type='ml.p3.2xlarge',
336+
instance_type='ml.g6.12xlarge',
337337
instance_count=1,
338338
role=role,
339339
transformers_version='4.26',

0 commit comments

Comments
 (0)