@@ -127,7 +127,7 @@ hyperparameters={'epochs': 1,
127127huggingface_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,
269269huggingface_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 = [
333333huggingface_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