Skip to content

Commit 7114c35

Browse files
author
Hui Wang
committed
Change default compile model max run to 15 mins
1 parent 934fd90 commit 7114c35

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

src/sagemaker/estimator.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -476,7 +476,7 @@ def compile_model(
476476
output_path,
477477
framework=None,
478478
framework_version=None,
479-
compile_max_run=5 * 60,
479+
compile_max_run=15 * 60,
480480
tags=None,
481481
**kwargs
482482
):

tests/unit/test_mxnet.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -160,7 +160,7 @@ def _create_compilation_job(input_shape, output_location):
160160
"job_name": COMPILATION_JOB_NAME,
161161
"output_model_config": {"S3OutputLocation": output_location, "TargetDevice": "ml_c4"},
162162
"role": ROLE,
163-
"stop_condition": {"MaxRuntimeInSeconds": 300},
163+
"stop_condition": {"MaxRuntimeInSeconds": 900},
164164
"tags": None,
165165
}
166166

0 commit comments

Comments
 (0)